from urllib.request
import urlretrieve
from os.path
import isfile, isdir
from tqdm
import tqdm
data_dir =
'data/'
if not isdir(data_dir):
raise Exception(
"Data directory doesn't exist!")
class DLProgress(tqdm):
last_block =
0
def hook(self, block_num=1, block_size=1, total_size=None):
self.total = total_size
self.update((block_num - self.last_block) * block_size)
self.last_block = block_num
if not isfile(data_dir +
"train_32x32.mat"):
with DLProgress(unit=
'B', unit_scale=
True, miniters=
1, desc=
'SVHN Training Set')
as pbar:
urlretrieve(
'http://ufldl.stanford.edu/housenumbers/train_32x32.mat',
data_dir +
'train_32x32.mat',
pbar.hook)
if not isfile(data_dir +
"test_32x32.mat"):
with DLProgress(unit=
'B', unit_scale=
True, miniters=
1, desc=
'SVHN Training Set')
as pbar:
urlretrieve(
'http://ufldl.stanford.edu/housenumbers/test_32x32.mat',
data_dir +
'test_32x32.mat',
pbar.hook)
转载请注明原文地址: https://www.6miu.com/read-96878.html
