Python ClassificationData示例

示例#1

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

示例#2

文件： __init__.py 项目： fishfishson/MeshCNN

def CreateDataset(opt):
    """loads dataset class"""

    if opt.dataset_mode == 'segmentation':
        from data.dataloader import MSDSurfTrainDataset
        dataset = MSDSurfTrainDataset(opt)
    elif opt.dataset_mode == 'classification':
        from data.classification_data import ClassificationData
        dataset = ClassificationData(opt)
    return dataset

示例#3

文件： __init__.py 项目： s183983/MeshCNN_sparse

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

示例#4

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

示例#5

文件： __init__.py 项目： Simon-Fuhaoyuan/MeshCNN

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