new dataset

MadryLab · Nov 1, 2019 · 410d421 · 410d421
1 parent 1ba7881
commit 410d421
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/robustness/tools/imagenet_helpers.py b/robustness/tools/imagenet_helpers.py
@@ -333,7 +333,20 @@ def common_superclass_wnid(group_name):
                      'n07555863', #food (24)
                      'n03405725', #furniture (21)
                      'n02469914', #primate (20)
-                   ]
+                   ],
+	'mid_12':  ['n02084071', #dog (100+)
+                      'n01503061', #bird (52)
+                      'n04576211', #wheeled vehicle
+                      'n01661091', #reptile, reptilian (36)
+                      'n02075296', #carnivore
+                      'n02159955', #insect (27)
+                      'n03800933', #musical instrument (26)
+                      'n07555863', #food (24)
+                      'n03419014', #garment (24)
+                      'n03405725', #furniture (21)
+                      'n02469914', #primate (20)
+                      'n02512053', #fish (16)
+                    ]
     }
 
     if group_name in common_groups: