From cadf19f91e707d6e15b7e863c858e88d186a9e94 Mon Sep 17 00:00:00 2001 From: Kuang Date: Thu, 12 Nov 2020 09:41:19 +0800 Subject: [PATCH] fix issue: missing background class --- docs/model_zoo/csv_tables/Segmentations/SS_MHP-V1.csv | 2 +- gluoncv/data/mhp.py | 7 ++++--- 2 files changed, 5 insertions(+), 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 index dd5c6ca990..e21c71d18f 100644 --- 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 `_,`log `_ +icnet_resnet50_mhpv1,ICNet [5]_,90.5,44.5,`shell script `_,`log `_ diff --git a/gluoncv/data/mhp.py b/gluoncv/data/mhp.py index 4b9406c1a6..d62b609c2a 100644 --- 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)