download_t2_dataset.py 文件源码

python
阅读 20 收藏 0 点赞 0 评论 0

项目:TanksAndTemples 作者: IntelVCL 项目源码 文件源码
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')
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号