def __init__(self, **kwargs):
self.num_classes = 2
self.palette = palette.get_voc_palette(self.num_classes)
super().__init__(**kwargs)
def __init__(self, num_classes, rgb_ext, ann_ext, **kwargs):
self.rgb_ext = rgb_ext
self.ann_ext = ann_ext
self.num_classes = num_classes
self.palette = palette.get_voc_palette(self.num_classes)
super(PlantDataset, self).__init__(**kwargs)
def __init__(self, **kwargs):
self.num_classes = 21
self.palette = palette.get_voc_palette(self.num_classes)
super(TESTAugDataset, self).__init__(**kwargs)
def __init__(self, **kwargs):
self.num_classes = 8
self.palette = palette.get_voc_palette(self.num_classes)
super(RISFastDataset, self).__init__(**kwargs)
def __init__(self, **kwargs):
self.num_classes = 21
self.palette = palette.get_voc_palette(self.num_classes)
# self.M = 5
super(VOCDataset, self).__init__(**kwargs)
def __init__(self, num_classes=14, **kwargs):
self.num_classes = num_classes
self.palette = palette.get_voc_palette(self.num_classes)
super(RSIDataset, self).__init__(**kwargs)
def __init__(self, **kwargs):
self.num_classes = 21
self.palette = palette.get_voc_palette(self.num_classes)
self.size=128
super(VOCAugDataset, self).__init__(**kwargs)
