def download_image_sets(pathname, scene, image_md5_dict, calc_md5):
scene_out_dir = pathname + 'image_sets'
download_file = scene + '.zip'
download_file_local = scene_out_dir + sep + scene + '.zip'
print('\ndownloading image set ' + download_file.split('/')[-1])
idd = id_download_dict[download_file]
download_file_from_google_drive(idd, download_file_local)
if(calc_md5):
h_md5 = generate_file_md5(download_file_local)
print('\nmd5 downloaded: ' + h_md5)
print('md5 original: ' + image_md5_dict[scene])
md5_check = h_md5 == image_md5_dict[scene]
if(md5_check):
if(unpack):
extr_dir = scene_out_dir
zip_file = scene_out_dir + sep + scene + '.zip'
if(zipfile.is_zipfile(zip_file)):
if not os.path.exists(extr_dir):
os.makedirs(extr_dir)
zip = zipfile.ZipFile(zip_file,'r')
zip.extractall(extr_dir)
else:
print('\nWarning: MD5 does not match, delete file and restart' +
' download\n')
评论列表
文章目录