File size: 2,689 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 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 | # 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 iMaterialistDataset(BaseSegDataset):
"""
iMaterialist 2019 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",
),
palette=[
[0, 0, 0],
[234, 191, 155], [186, 99, 123], [46, 100, 157], [154, 71, 196],
[15, 185, 171], [13, 89, 100], [67, 216, 41], [212, 139, 166],
[10, 101, 73], [198, 51, 168], [38, 174, 154], [150, 192, 158],
[194, 243, 120], [10, 224, 173], [214, 94, 149], [211, 126, 18],
[96, 7, 165], [255, 35, 14], [83, 127, 78], [106, 23, 51],
[41, 244, 224], [38, 86, 244], [244, 234, 150], [233, 247, 180],
[222, 117, 26], [2, 90, 51], [27, 176, 90], [178, 160, 25],
[75, 52, 236], [119, 65, 186], [163, 254, 113], [39, 140, 118],
[235, 112, 193], [134, 107, 77], [57, 169, 93], [251, 104, 47],
[224, 14, 49], [20, 123, 134], [178, 32, 212], [116, 194, 248],
[211, 196, 233], [93, 36, 29], [113, 99, 55], [5, 7, 250],
[172, 174, 41], [101, 98, 209],
],)
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)
|