Python KeyDataset示例

示例#1

文件： custom_dataset_data_loader.py 项目： rocketappslab/SCA-GAN

def CreateDataset(opt):
    from data.keypoint import KeyDataset
    dataset = KeyDataset()

    print("dataset [%s] was created" % (dataset.name()))
    dataset.initialize(opt)
    return dataset

示例#2

文件： custom_dataset_data_loader.py 项目： peternara/BiGraphGAN-human-body

def CreateDataset(opt):
    dataset = None

    if opt.dataset_mode == 'keypoint':
        from data.keypoint import KeyDataset
        dataset = KeyDataset()

    else:
        raise ValueError("Dataset [%s] not recognized." % opt.dataset_mode)

    print("dataset [%s] was created" % (dataset.name()))
    dataset.initialize(opt)
    return dataset

示例#3

文件： custom_dataset_data_loader.py 项目： shilongshen/ADGAN-explain

def CreateDataset(opt):
    dataset = None

    if opt.dataset_mode == 'keypoint':  #选这个，dataset = KeyDataset()
        from data.keypoint import KeyDataset
        dataset = KeyDataset()  #返回 #P1，BP1,SP1,P2,BP2,P1_name,P2_name
    # elif opt.dataset_mode == 'keypoint_mix':
    #     from data.keypoint_mix import KeyDataset
    #     dataset = KeyDataset()
    else:
        raise ValueError("Dataset [%s] not recognized." % opt.dataset_mode)

    print("dataset [%s] was created" % (dataset.name()))
    dataset.initialize(opt)
    return dataset