[Tensorflow] 使用 tf.keras.utils.get_file() 下载 MS-COCO 2014 数据集

 import argparse

 import tensorflow as tf
 tf.enable_eager_execution()

 def main(args):
     """Download the Microsoft COCO 2014 data set."""
     # Annotation zip
     tf.keras.utils.get_file(fname=args.annotation_zip,
                             origin=args.annotation_origin,
                             cache_dir=args.cache_dir,
                             cache_subdir=args.cache_subdir,
                             extract=True)

     # Train image zip
     tf.keras.utils.get_file(fname=args.train_zip,
                             origin=args.train_origin,
                             cache_dir=args.cache_dir,
                             cache_subdir=args.cache_subdir,
                             extract=True)

     # Val image zip
     tf.keras.utils.get_file(fname=args.val_zip,
                             origin=args.val_origin,
                             cache_dir=args.cache_dir,
                             cache_subdir=args.cache_subdir,
                             extract=True)

     # Test image zip
     tf.keras.utils.get_file(fname=args.test_zip,
                             origin=args.test_origin,
                             cache_dir=args.cache_dir,
                             cache_subdir=args.cache_subdir,
                             extract=True)

 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
     parser.add_argument('--cache_dir', type=str, default='F:/', help='location to store cached files')
     parser.add_argument('--cache_subdir', type=str, default='COCO2014/', help='subdirectory under the cache dir')
     parser.add_argument('--annotation_zip', type=str, default='annotation2014.zip', help='name of annotation zip')
     parser.add_argument('--annotation_origin', type=str,
                         default='http://images.cocodataset.org/annotations/annotations_trainval2014.zip',
                         help='origin of annotation zip')
     parser.add_argument('--train_zip', type=str, default='train2014.zip', help='name of train zip')
     parser.add_argument('--train_origin', type=str,
                         default='http://images.cocodataset.org/zips/train2014.zip',
                         help='origin of train zip')
     parser.add_argument('--val_zip', type=str, default='val2014.zip', help='name of val zip')
     parser.add_argument('--val_origin', type=str,
                         default='http://images.cocodataset.org/zips/val2014.zip',
                         help='origin of val zip')
     parser.add_argument('--test_zip', type=str, default='test2014.zip', help='name of test zip')
     parser.add_argument('--test_origin', type=str,
                         default='http://images.cocodataset.org/zips/test2014.zip',
                         help='origin of test zip')

     args = parser.parse_args()
     print(args)

     main(args)