forked from tensorlayer/DCGAN
-
Notifications
You must be signed in to change notification settings - Fork 0
/
bird_data.py
43 lines (38 loc) · 1.04 KB
/
bird_data.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
'''
download bird dataset from : http://www.vision.caltech.edu/visipedia/CUB-200-2011.html
'''
from __future__ import print_function
from six.moves import urllib
import requests
# import urllib2
import os
import sys
cur_dir = os.getcwd()
image_dir = os.path.join(cur_dir,"data/bird.tgz")
def download(url, dirpath):
filepath = dirpath
u = urllib.request.urlopen(url)
f = open(filepath, 'wb')
filesize = int(u.headers["Content-Length"])
print("Downloading: %s Bytes: %s" % ("bird", filesize))
downloaded = 0
block_sz = 8192
status_width = 70
while True:
buf = u.read(block_sz)
if not buf:
print('')
break
else:
print('', end='\r')
downloaded += len(buf)
f.write(buf)
status = (("[%-" + str(status_width + 1) + "s] %3.2f%%") %
('=' * int(float(downloaded) / filesize * status_width) + '>', downloaded * 100. / filesize))
print(status, end='')
sys.stdout.flush()
f.close()
return filepath
if __name__ == '__main__':
url = "http://www.vision.caltech.edu/visipedia-data/CUB-200-2011/CUB_200_2011.tgz"
download(url,image_dir)