Update imagenet_classes.json

imagenet_classes.json

@@ -1,15 +1,12 @@
-{
-"0": "tench",
-"1": "goldfish",
-"2": "great white shark",
-"3": "tiger shark",
-"4": "hammerhead shark",
-"5": "electric ray",
-"6": "stingray",
-"7": "cock",
-"8": "hen",
-"9": "ostrich",
-"10": "brambling",
-"...": "...",
-"999": "toilet tissue"
-}
+import json
+
+# Download from official repo
+!wget -q https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt
+
+with open("imagenet_classes.txt") as f:
+    classes = f.read().splitlines()
+
+mapping = {str(i): classes[i] for i in range(len(classes))}
+
+with open("imagenet_classes.json", "w") as f:
+    json.dump(mapping, f)