1. 程式人生 > >深度學習(1)—從ImageNet下載圖片

深度學習(1)—從ImageNet下載圖片

當我們訓練網路時需要大量的圖片資料,我們可以去開源的imagenet庫下載圖片,首先我們需要註冊ImageNet賬號,郵箱需要.edu結尾的郵箱或者別的公司的郵箱,免費的郵箱是不行的。註冊還需要翻牆,否則看不到reCAPTCHA驗證碼。註冊後再登入搜尋需要的圖片標註,比如說dog。然後選擇下載連結存在.txt檔案裡,最後就是編寫Python3指令碼編寫啦

from urllib import request
#import urllib2, urllib
import signal

path ='C:/Users/Administrator/Desktop/text.txt'  #儲存的.txt檔案
paths='C:/Users/Administrator/Desktop/r_g_pic/' 圖片下載儲存的地址


file = open(path,'r')
i=1004
for line in file:
    try:
        paths='C:/Users/Administrator/Desktop/r_g_pic/'
        paths=paths+str(i)+'.jpg'
        request.urlretrieve(line,paths)
        print(i)
        i=i+1
    except:
        print("%s timeout " % line)
        pass

file.close()

你也可以使用多執行緒下載這樣會下載更快。完成後就可以看見很多圖片啦!!!