icnet missing background class (#1518)

dmlc · Nov 12, 2020 · d4f2893 · d4f2893
1 parent 9f6578e
commit d4f2893
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/docs/model_zoo/csv_tables/Segmentations/SS_MHP-V1.csv b/docs/model_zoo/csv_tables/Segmentations/SS_MHP-V1.csv
@@ -1,2 +1,2 @@
 Name,Method,pixAcc,mIoU,Command,log
-icnet_resnet50_mhpv1,ICNet [5]_,89.5,40.4,`shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/icnet_resnet50_mhpv1.sh>`_,`log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/icnet_resnet50_mhpv1.log>`_
+icnet_resnet50_mhpv1,ICNet [5]_,90.5,44.5,`shell script <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/icnet_resnet50_mhpv1.sh>`_,`log <https://raw.githubusercontent.com/dmlc/web-data/master/gluoncv/logs/segmentation/icnet_resnet50_mhpv1.log>`_
diff --git a/gluoncv/data/mhp.py b/gluoncv/data/mhp.py
@@ -36,8 +36,8 @@ class MHPV1Segmentation(SegmentationDataset):
     >>>     num_workers=4)
     """
     # pylint: disable=abstract-method
-    NUM_CLASS = 18
-    CLASSES = ("hat", "hair", "sunglasses", "upper clothes", "skirt",
+    NUM_CLASS = 19
+    CLASSES = ("background", "hat", "hair", "sunglasses", "upper clothes", "skirt",
                "pants", "dress", "belt", "left shoe", "right shoe", "face", "left leg",
                "right leg", "left arm", "right arm", "bag", "scarf", "torso skin")
 
@@ -99,7 +99,8 @@ def __getitem__(self, index):
         return img, mask
 
     def _mask_transform(self, mask):
-        return mx.nd.array(np.array(mask), mx.cpu(0)).astype('int32')   # - 1
+        target = np.array(mask).astype('int32')
+        return mx.nd.array(target, mx.cpu(0))
 
     def __len__(self):
         return len(self.images)