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',

60
mmseg/datasets/__init__.py
View File

@ -1,30 +1,30 @@
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
from .ade import ADE20KDataset from .ade import ADE20KDataset
from .builder import DATASETS, PIPELINES, build_dataloader, build_dataset from .builder import DATASETS, PIPELINES, build_dataloader, build_dataset
from .chase_db1 import ChaseDB1Dataset from .chase_db1 import ChaseDB1Dataset
from .cityscapes import CityscapesDataset from .cityscapes import CityscapesDataset
from .coco_stuff import COCOStuffDataset from .coco_stuff import COCOStuffDataset
from .custom import CustomDataset from .custom import CustomDataset
from .dark_zurich import DarkZurichDataset from .dark_zurich import DarkZurichDataset
from .dataset_wrappers import (ConcatDataset, MultiImageMixDataset, from .dataset_wrappers import (ConcatDataset, MultiImageMixDataset,
RepeatDataset) RepeatDataset)
from .drive import DRIVEDataset from .drive import DRIVEDataset
from .hrf import HRFDataset from .hrf import HRFDataset
from .isaid import iSAIDDataset from .isaid import iSAIDDataset
from .isprs import ISPRSDataset from .isprs import ISPRSDataset
from .loveda import LoveDADataset from .loveda import LoveDADataset
from .night_driving import NightDrivingDataset from .night_driving import NightDrivingDataset
from .pascal_context import PascalContextDataset, PascalContextDataset59 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',
'DATASETS', 'build_dataset', 'PIPELINES', 'CityscapesDataset', 'DATASETS', 'build_dataset', 'PIPELINES', 'CityscapesDataset',
'PascalVOCDataset', 'ADE20KDataset', 'PascalContextDataset', 'PascalVOCDataset', 'ADE20KDataset', 'PascalContextDataset',
'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']

46
mmseg/datasets/face.py
View File

@ -1,23 +1,23 @@
# Copyright (c) OpenMMLab. All rights reserved. # Copyright (c) OpenMMLab. All rights reserved.
import os.path as osp import os.path as osp
from .builder import DATASETS from .builder import DATASETS
from .custom import CustomDataset from .custom import CustomDataset
@DATASETS.register_module() @DATASETS.register_module()
class FaceOccluded(CustomDataset): class FaceOccludedDataset(CustomDataset):
"""Face Occluded dataset. """Face Occluded dataset.
Args: Args:
split (str): Split txt file for Pascal VOC. split (str): Split txt file for Pascal VOC.
""" """
CLASSES = ('background', 'face') CLASSES = ('background', 'face')
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