Spaces:
Runtime error
Runtime error
Update main.py
Browse files
main.py
CHANGED
@@ -67,7 +67,6 @@ You will use the [MS-COCO dataset](http://cocodataset.org/#home) to train your m
|
|
67 |
|
68 |
**Caution: large download ahead**. You'll use the training set, which is a 13GB file.
|
69 |
"""
|
70 |
-
# os.system("wget http://images.cocodataset.org/zips/train2014.zip")
|
71 |
# Download caption annotation files
|
72 |
annotation_folder = '/annotations/'
|
73 |
annotation_zip = tf.keras.utils.get_file('captions.zip',
|
@@ -78,13 +77,14 @@ annotation_file = os.path.dirname(annotation_zip)+'/annotations/captions_train20
|
|
78 |
# os.remove(annotation_zip)
|
79 |
|
80 |
# Download image files
|
|
|
81 |
image_folder = '/train2014/'
|
82 |
|
83 |
-
image_zip = tf.keras.utils.get_file('train2014.zip',
|
84 |
-
|
85 |
-
|
86 |
-
|
87 |
-
PATH = os.path.dirname(
|
88 |
# os.remove(image_zip)
|
89 |
|
90 |
|
|
|
67 |
|
68 |
**Caution: large download ahead**. You'll use the training set, which is a 13GB file.
|
69 |
"""
|
|
|
70 |
# Download caption annotation files
|
71 |
annotation_folder = '/annotations/'
|
72 |
annotation_zip = tf.keras.utils.get_file('captions.zip',
|
|
|
77 |
# os.remove(annotation_zip)
|
78 |
|
79 |
# Download image files
|
80 |
+
os.system("wget http://images.cocodataset.org/zips/train2014.zip")
|
81 |
image_folder = '/train2014/'
|
82 |
|
83 |
+
# image_zip = tf.keras.utils.get_file('train2014.zip',
|
84 |
+
# cache_subdir=os.path.abspath('.'),
|
85 |
+
# origin='http://images.cocodataset.org/zips/train2014.zip',
|
86 |
+
# extract=True)
|
87 |
+
PATH = os.path.dirname("train2014.zip") + image_folder
|
88 |
# os.remove(image_zip)
|
89 |
|
90 |
|