Merge pull request #14 from fpjentzsch/dev

Fix: Sorting of labeled ImageNet subfolders
Xilinx · Aug 27, 2021 · 61febea · 61febea
2 parents ce4e090 + f9d44ea
commit 61febea
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/finn_examples/notebooks/2_imagenet_with_cnns.ipynb b/finn_examples/notebooks/2_imagenet_with_cnns.ipynb
@@ -166,6 +166,7 @@
     "def setup_dataloader(val_path, label_file_path = None, batch_size=100, n_images = 50000):\n",
     "    if label_file_path is None:\n",
     "        val_folders = [ f.name for f in os.scandir(val_path) if f.is_dir() ]\n",
+    "        val_folders = sorted(val_folders)\n",
     "        assert len(val_folders) == 1000, \"Expected 1000 subfolders in ILSVRC2012 val\"\n",
     "        files = []\n",
     "        labels = []\n",