Function get_tex_model_list

pointersect/data/genlist.py:19–49

given setting (train/test), generate a list of obj file names in the tex dataset Args: setting: train/test num_classes: number of class in training rnd_seed: random seed Returns: a list of filenames

(
        setting: str,
        num_classes: int = 3,
        rnd_seed: int = 0
)

# randomly pick mesh files from datasets

def get_tex_model_list(
        setting: str,
        num_classes: int = 3,
        rnd_seed: int = 0
) -> list:
    """
    given setting (train/test), generate a list of obj file names in the tex dataset
    Args:
        setting: train/test
        num_classes: number of class in training
        rnd_seed: random seed

    Returns:
        a list of filenames

    """
    class_list = ['buddha', 'buddha2', 'bunny', 'cat', 'feline', 'tiger', 'zebra']

    if rnd_seed > 0:  # negative: use sorted class_list
        np.random.seed(rnd_seed)
        class_list = np.random.permutation(class_list).tolist()

    if setting == 'train':
        class_list = class_list[:num_classes]
    elif setting == 'test':
        class_list.reverse()
        class_list = class_list[:num_classes]

    obj_path_list = [c + '.obj' for c in class_list]

    return obj_path_list


def get_ShapeNet_model_list(