mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

fix format

Browse Source
This commit is contained in:
jinwonkim93 2022-10-17 01:47:59 +00:00
parent 33156d0e94
commit 1ca1780f5a
4 changed files with 68 additions and 68 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
configs/_base_/datasets/occlude_face.py
View File

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

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

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

4
mmseg/datasets/__init__.py
View File

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

4
mmseg/datasets/face.py
View File

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