Make the labels of imagenet unique. (#4641)

6b0097b7 · Vasilis Vryniotis · GitHub · ec1bb0f6 · 6b0097b7
Unverified Commit 6b0097b7 authored Oct 18, 2021 by Vasilis Vryniotis Committed by GitHub Oct 18, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

torchvision/prototype/models/_meta.py torchvision/prototype/models/_meta.py +2 -2

No files found.
--- a/torchvision/prototype/models/_meta.py
+++ b/torchvision/prototype/models/_meta.py
@@ -139,7 +139,7 @@ _IMAGENET_CATEGORIES = [
    "little blue heron",
    "American egret",
    "bittern",
-    "crane",
+    "crane bird",
    "limpkin",
    "European gallinule",
    "American coot",
@@ -644,7 +644,7 @@ _IMAGENET_CATEGORIES = [
    "mailbag",
    "mailbox",
    "maillot",
-    "maillot",
+    "maillot tank suit",
    "manhole cover",
    "maraca",
    "marimba",