Some questions about the data load.

[RiverZhenjiang]

it seems that the 'test' and 'trainval' of trian.py is not fit the data loader.

train.py

PCN_TRAIN_DATASET, PCN_TEST_DATASET, AE_TRAIN_DATASET, AE_TEST_DATASET, NUM_PARTS = data_utils.load_data(DATA_PATH,
                                                                                                         NUM_POINTS,
                                                                                                         CATEGORY,
                                                                                                         'test',
                                                                                                         'trainval')

data_utils.py

def load_data(data_path, num_point, category, seen_split, unseen_split):
    pcn_train_dataset, pcn_test_dataset, num_parts = load_pcn_data(data_path, num_point, category, seen_split,
                                                                   unseen_split)
    ae_train_dataset, ae_test_dataset = load_aes_data(data_path, num_point, category, seen_split, unseen_split,
                                                      num_parts)

    return pcn_train_dataset, pcn_test_dataset, ae_train_dataset, ae_test_dataset, num_parts


def load_pcn_data(data_path, num_point, category, seen_split, unseen_split):
    pcn_train_dataset = part_dataset_pcn.PartDatasetPCN(root=data_path, npoints=num_point, class_choice=category,
                                                        split=seen_split)

    pcn_test_dataset = part_dataset_pcn.PartDatasetPCN(root=data_path, npoints=num_point, class_choice=category,
                                                       split=unseen_split)
    num_parts = pcn_train_dataset.get_number_of_parts()

    return pcn_train_dataset, pcn_test_dataset, num_parts