def CreateDataset(opt): """loads dataset class""" if opt.dataset_mode == 'segmentation': from data.segmentation_data import SegmentationData dataset = SegmentationData(opt) elif opt.dataset_mode == 'classification': from data.classification_data import ClassificationData dataset = ClassificationData(opt) elif opt.dataset_mode == 'distance_field': from data.segmentation_data import SegmentationData dataset = SegmentationData(opt) return dataset
def CreateDataset(opt): """loads dataset class""" if opt.dataset_mode == 'segmentation': from data.segmentation_data import SegmentationData dataset = SegmentationData(opt) elif opt.dataset_mode in ('classification', 'regression', 'binary_class'): from data.classification_data import ClassificationData dataset = ClassificationData(opt) return dataset
def CreateDataset(opt): """loads dataset class""" if opt.dataset_mode == 'segmentation': from data.segmentation_data import SegmentationData dataset = SegmentationData(opt) elif opt.dataset_mode == 'classification': from data.classification_data import ClassificationData dataset = ClassificationData(opt) elif opt.dataset_mode == 'texturize': from data.texturize_data import TexturizeData dataset = TexturizeData(opt) return dataset
def CreateDataset(opt): """loads dataset class""" if opt.dataset_mode == 'segmentation': from data.segmentation_data import SegmentationData dataset = SegmentationData(opt) elif opt.dataset_mode == 'classification': from data.classification_data import ClassificationData dataset = ClassificationData(opt) elif opt.dataset_mode == 'reconstruction': from data.reconstruction_data import ReconstructionData dataset = ReconstructionData(opt) return dataset