Skip to content

Commit 6e0a96d

Browse files
committed
add dataset class file for feet segmentation
1 parent f5b4f76 commit 6e0a96d

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

mmseg/datasets/my_feetseg.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,13 @@
77

88
@DATASETS.register_module()
99
class MyFeetSegDataset(CustomDataset):
10-
"""feet segmentation dataset. 0 stands for background, 1 for left foot, 2 for right foot
11-
which is included in 3 categories.
10+
"""feet segmentation dataset. 0 stands for background, 1 for foot
11+
which is included in 2 categories.
1212
"""
1313

14-
CLASSES = ('background','left foot', 'right foot')
14+
CLASSES = ('background','feet')
1515

16-
PALETTE = [[120,120,120],[128, 64, 128], [192, 96, 128]]
16+
PALETTE = [[120,120,120],[255, 64, 128]]
1717

1818
def __init__(self,
1919
img_suffix='.jpg',

0 commit comments

Comments
 (0)