Spaces:
Starting
on
A10G
Starting
on
A10G
import os.path as osp | |
from .builder import DATASETS | |
from .custom import CustomDataset | |
class HRFDataset(CustomDataset): | |
"""HRF dataset. | |
In segmentation map annotation for HRF, 0 stands for background, which is | |
included in 2 categories. ``reduce_zero_label`` is fixed to False. The | |
``img_suffix`` is fixed to '.png' and ``seg_map_suffix`` is fixed to | |
'.png'. | |
""" | |
CLASSES = ('background', 'vessel') | |
PALETTE = [[120, 120, 120], [6, 230, 230]] | |
def __init__(self, **kwargs): | |
super(HRFDataset, self).__init__( | |
img_suffix='.png', | |
seg_map_suffix='.png', | |
reduce_zero_label=False, | |
**kwargs) | |
assert osp.exists(self.img_dir) | |