fix format

configs/_base_/datasets/occlude_face.py

@@ -1,5 +1,5 @@
-dataset_type = 'FaceOccluded'
-data_root = 'data/occlusion-aware-dataset'
+dataset_type = 'FaceOccludedDataset'
+data_root = 'data/occlusion-aware-face-dataset'
 crop_size = (512, 512)
 img_norm_cfg = dict(
     mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375], to_rgb=True)
@@ -40,15 +40,15 @@ test_pipeline = [
 ]
 
 dataset_train_A = dict(
-    type='FaceOccluded',
+    type=dataset_type,
     data_root=data_root,
     img_dir='CelebAMask-HQ-original/image',
     ann_dir='CelebAMask-HQ-original/mask_edited',
-    split='CelebAMask-HQ-original/split/train_ori.txt',
+    split='CelebAMask-HQ-original/split/train.txt',
     pipeline=train_pipeline)
 
 dataset_train_B = dict(
-    type='FaceOccluded',
+    type=dataset_type,
     data_root=data_root,
     img_dir='NatOcc-SOT/image',
     ann_dir='NatOcc-SOT/mask',
@@ -57,26 +57,26 @@ dataset_train_B = dict(
 
 
 dataset_valid = dict(
-        type='FaceOccluded',
+        type=dataset_type,
         data_root=data_root,
-        img_dir='occlusion-aware-dataset/HQ-FO-dataset/RealOcc/image',
-        ann_dir='occlusion-aware-dataset/HQ-FO-dataset/RealOcc/mask',
-        split='occlusion-aware-dataset/HQ-FO-dataset/RealOcc/split/val.txt',
+        img_dir='RealOcc/image',
+        ann_dir='RealOcc/mask',
+        split='RealOcc/split/val.txt',
         pipeline=test_pipeline)
 
 dataset_test = dict(
-        type='FaceOccluded',
+        type=dataset_type,
         data_root=data_root,
-        img_dir='occlusion-aware-dataset/HQ-FO-dataset/RealOcc/image',
-        ann_dir='occlusion-aware-dataset/HQ-FO-dataset/RealOcc/mask',
-        split='occlusion-aware-dataset/HQ-FO-dataset/RealOcc/test.txt',
+        img_dir='RealOcc/image',
+        ann_dir='RealOcc/mask',
+        split='RealOcc/test.txt',
         pipeline=test_pipeline)
 
 data = dict(
     samples_per_gpu=2,
     workers_per_gpu=2,
     train=[
-            dataset_train_A,dataset_train_B,
+            dataset_train_A,dataset_train_B
     ],
     val= dataset_valid,
     test=dataset_test)

configs/deeplabv3plus/deeplabv3plus_r101_512x512_C-CM+C-WO-NatOcc-SOT.py

@@ -1,5 +1,5 @@
 # +
-_base_ = '../_base_/dataset/occlude_face.py'
+_base_ = '../_base_/datasets/occlude_face.py'
 norm_cfg = dict(type='SyncBN', requires_grad=True)
 model = dict(
     type='EncoderDecoder',

mmseg/datasets/__init__.py

@@ -18,7 +18,7 @@ from .pascal_context import PascalContextDataset, PascalContextDataset59
 from .potsdam import PotsdamDataset
 from .stare import STAREDataset
 from .voc import PascalVOCDataset
-from .face import FaceOccluded
+from .face import FaceOccludedDataset
 
 __all__ = [
     'CustomDataset', 'build_dataloader', 'ConcatDataset', 'RepeatDataset',
@@ -27,4 +27,4 @@ __all__ = [
     'PascalContextDataset59', 'ChaseDB1Dataset', 'DRIVEDataset', 'HRFDataset',
     'STAREDataset', 'DarkZurichDataset', 'NightDrivingDataset',
     'COCOStuffDataset', 'LoveDADataset', 'MultiImageMixDataset',
-    'iSAIDDataset', 'ISPRSDataset', 'PotsdamDataset',"FaceOccluded"]
+    'iSAIDDataset', 'ISPRSDataset', 'PotsdamDataset', 'FaceOccludedDataset']

mmseg/datasets/face.py

@@ -6,7 +6,7 @@ from .custom import CustomDataset
 
 
 @DATASETS.register_module()
-class FaceOccluded(CustomDataset):
+class FaceOccludedDataset(CustomDataset):
     """Face Occluded dataset.
 
     Args:
@@ -18,6 +18,6 @@ class FaceOccluded(CustomDataset):
     PALETTE = [[0, 0, 0], [128, 0, 0]]
 
     def __init__(self, split, **kwargs):
-        super(FaceOccluded, self).__init__(
+        super(FaceOccludedDataset, self).__init__(
             img_suffix='.jpg', seg_map_suffix='.png', split=split, **kwargs)
         assert osp.exists(self.img_dir) and self.split is not None