fixed bug: the imagenet index has to start from 0

eljanmahammadli · Nov 20, 2023 · 7334559 · 7334559
1 parent 286daca
commit 7334559
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/imagenet.py b/examples/imagenet.py
@@ -1000,4 +1000,4 @@
     "ear",
     "toilet tissue",
 ]
-IMAGENET_CATEGORIES = {i + 1: v for i, v in enumerate(IMAGENET_CATEGORIES)}
+IMAGENET_CATEGORIES = {i: v for i, v in enumerate(IMAGENET_CATEGORIES)}