Returns: two list of mesh filenames (for training and test)
(
dataset_name: str,
)
| 32 | |
| 33 | |
| 34 | def get_mesh_filenames( |
| 35 | dataset_name: str, |
| 36 | ) -> (list, list): |
| 37 | """ |
| 38 | Returns: |
| 39 | two list of mesh filenames (for training and test) |
| 40 | """ |
| 41 | |
| 42 | mesh_filename_dict = get_defualt_mesh_filename_config() |
| 43 | |
| 44 | if dataset_name.lower() == 'tex': |
| 45 | train_filename = \ |
| 46 | get_tex_model_list( |
| 47 | setting='train', |
| 48 | num_classes=3, |
| 49 | rnd_seed=-1 |
| 50 | ) |
| 51 | test_filename = \ |
| 52 | get_tex_model_list( |
| 53 | setting='test', |
| 54 | num_classes=3, |
| 55 | rnd_seed=-1 |
| 56 | ) |
| 57 | dataset_root_dir = 'datasets/tex-models' |
| 58 | elif dataset_name.lower() == 'shapenet': |
| 59 | train_filename = mesh_filename_dict['shapenet']['train'] |
| 60 | test_filename = mesh_filename_dict['shapenet']['test'] |
| 61 | dataset_root_dir = mesh_filename_dict['shapenet']['dataset_root_dir'] |
| 62 | elif dataset_name.lower() == 'sketchfab': |
| 63 | train_filename = mesh_filename_dict['sketchfab']['train'] |
| 64 | test_filename = mesh_filename_dict['sketchfab']['test'] |
| 65 | dataset_root_dir = mesh_filename_dict['sketchfab']['dataset_root_dir'] |
| 66 | elif dataset_name.lower() == 'sketchfab-small': |
| 67 | train_filename = mesh_filename_dict['sketchfab-small']['train'] |
| 68 | test_filename = mesh_filename_dict['sketchfab-small']['test'] |
| 69 | dataset_root_dir = mesh_filename_dict['sketchfab-small']['dataset_root_dir'] |
| 70 | else: |
| 71 | raise NotImplementedError |
| 72 | |
| 73 | return train_filename, test_filename, dataset_root_dir |
| 74 | |
| 75 | |
| 76 | def set_dataset( |
no test coverage detected