使用gdown下载google drive数据集

做深度学习模型训练时,我们有时需要在Google drive下载数据集。数据集一般都是很大的file,我们可以使用gdown这个library进行下载。

假设我要下载DIOR这个dataset。首先,找到相应的Google driveDIOR dataset on google drive

假设我要下载Annotations.zip。右键点击file,点击copy link。在这个网站产生一条可以直接下载的链接。

使用以下python代码:

import gdown
url = "https://drive.google.com/uc?export=download&id=1KoQzqR20qvIXDf1qsXCHGxD003IPmXMw"
output = "Annotations.zip"
gdown.download(url, output)

这样就能把Annotations.zip从Google drive下载下来。如果你使用curl/wget下载大数据集,容易因为安全警告而失败。

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐