File size: 2,958 Bytes
6ed4a9c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | # Copyright (c) OpenMMLab. All rights reserved.
import os.path as osp
import mmengine.fileio as fileio
from mmseg.registry import DATASETS
from .basesegdataset import BaseSegDataset
@DATASETS.register_module()
class HumanParsingDataset(BaseSegDataset):
"""
Human Parsing dataset.
"""
METAINFO = dict(
classes=(
'Background', 'shirt, blouse', 'top, t-shirt, sweatshirt', 'sweater', 'cardigan', 'jacket', 'vest', 'pants', 'shorts', 'skirt', 'coat', 'dress', 'jumpsuit', 'cape', 'glasses', 'hat', 'headband, head covering, hair accessory', 'tie', 'glove', 'watch', 'belt', 'leg warmer', 'tights, stockings', 'sock', 'shoe', 'bag, wallet', 'scarf', 'umbrella', 'hood', 'collar', 'lapel', 'epaulette', 'sleeve', 'pocket', 'neckline', 'buckle', 'zipper', 'applique', 'bead', 'bow', 'flower', 'fringe', 'ribbon', 'rivet', 'ruffle', 'sequin', 'tassel', 'Hair', 'Sunglasses', 'Upper-clothes', 'Left-shoe', 'Right-shoe', 'Face', 'Left-leg', 'Right-leg', 'Left-arm', 'Right-arm'),
palette=[
[0, 0, 0], [255, 0, 0], [0, 128, 0], [0, 0, 255],
[0, 128, 128], [238, 130, 238], [128, 128, 128], [255, 255, 0],
[255, 153, 18], [255, 125, 64], [127, 255, 0], [175, 238, 238],
[138, 43, 226], [210, 105, 30], [0, 0, 139], [72, 61, 139],
[255, 20, 147], [255, 192, 203], [205, 92, 92], [32, 178, 170],
[132, 112, 255], [160, 82, 45], [255, 222, 173],
[240, 230, 140], [173, 216, 230], [250, 128, 114], [107, 142, 35],
[72, 209, 204], [199, 21, 133], [25, 25, 112], [123, 104, 238],
[0, 250, 154], [34, 139, 34], [219, 112, 147], [240, 128, 128],
[143, 188, 143], [47, 79, 79], [188, 143, 143], [100, 149, 237],
[102, 205, 170], [255, 160, 122], [147, 112, 219], [60, 179, 113],
[139, 0, 139], [255, 215, 0], [233, 150, 122], [0, 206, 209],
[148, 0, 211], [144, 238, 144], [255, 105, 180], [30, 144, 255],
[255, 140, 0], [153, 50, 204], [220, 20, 60], [46, 139, 87],
[240, 230, 155], [255, 99, 71]
])
# palette=[
# [0, 0, 0], [0, 192, 64], [0, 64, 96], [128, 192, 192],
# [0, 64, 64], [0, 192, 224], [0, 192, 192], [128, 192, 64],
# [0, 192, 96], [128, 192, 64], [128, 32, 192], [0, 0, 224],
# [0, 0, 64], [0, 160, 192], [128, 0, 96], [128, 0, 192],
# [0, 32, 192], [128, 128, 224], [0, 0, 192], [128, 160, 192],
# [128, 128, 0], [128, 0, 32], [128, 32, 0], [128, 0, 128],
# ],)
def __init__(self,
# ann_file,
img_suffix='.jpg',
seg_map_suffix='.png',
**kwargs) -> None:
super().__init__(
img_suffix=img_suffix,
seg_map_suffix=seg_map_suffix,
**kwargs)
|