revert change for colab

osfa · Feb 21, 2020 · 5a3c8d7 · 5a3c8d7
1 parent c876ad1
commit 5a3c8d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dataset_tool.py b/dataset_tool.py
@@ -691,7 +691,7 @@ def create_from_images(tfrecord_dir, image_dir, shuffle, res_log2=7, resize=None
 
 def create_from_images_raw(tfrecord_dir, image_dir, shuffle, res_log2=7, resize=None):
     print('Loading images from "%s"' % image_dir)
-    image_filenames = _get_all_files_2(image_dir)
+    image_filenames = _get_all_files(image_dir)
     print(f"detected {len(image_filenames)} images ...")
     if len(image_filenames) == 0:
         error("No input images found")