OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
16 lines
642 B
16 lines
642 B
# Copyright (c) OpenMMLab. All rights reserved. |
|
from .builder import DATASETS |
|
from .coco import CocoDataset |
|
|
|
|
|
@DATASETS.register_module() |
|
class DeepFashionDataset(CocoDataset): |
|
|
|
CLASSES = ('top', 'skirt', 'leggings', 'dress', 'outer', 'pants', 'bag', |
|
'neckwear', 'headwear', 'eyeglass', 'belt', 'footwear', 'hair', |
|
'skin', 'face') |
|
|
|
PALETTE = [(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, 32, 192), (0, 0, 224), (0, 0, 64), (0, 160, 192), |
|
(128, 0, 96), (128, 0, 192), (0, 32, 192)]
|
|
|