#578 Feature/sg 516 support head replacement for local pretrained weights unknown dataset

Merged

Ghost merged 1 commits into Deci-AI:master from deci-ai:feature/SG-516_support_head_replacement_for_local_pretrained_weights_unknown_dataset

  
1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

            from super_gradients.training.datasets.segmentation_datasets.cityscape_segmentation import CityscapesDataset
from super_gradients.training.datasets.segmentation_datasets.coco_segmentation import CoCoSegmentationDataSet
from super_gradients.training.datasets.segmentation_datasets.pascal_voc_segmentation import (
    PascalVOC2012SegmentationDataSet,
    PascalVOCAndAUGUnifiedDataset,
    PascalAUG2012SegmentationDataSet,
)
from super_gradients.training.datasets.segmentation_datasets.segmentation_dataset import SegmentationDataSet
from super_gradients.training.datasets.segmentation_datasets.supervisely_persons_segmentation import SuperviselyPersonsDataset
__all__ = [
    "SegmentationDataSet",
    "CoCoSegmentationDataSet",
    "PascalAUG2012SegmentationDataSet",
    "PascalVOC2012SegmentationDataSet",
    "CityscapesDataset",
    "SuperviselyPersonsDataset",
    "PascalVOCAndAUGUnifiedDataset",
]

Tip!

Press p or to see the previous file or, n or to see the next file

Deci-AI / super-gradients connected to https://github.com/Deci-AI/super-gradients.git

#578 Feature/sg 516 support head replacement for local pretrained weights unknown dataset

Deci-AI
/
super-gradients
connected to https://github.com/Deci-AI/super-gradients.git