(file_dir, cat_shape_dict)
| 26 | from pytorch3d.transforms import quaternion_to_matrix |
| 27 | |
| 28 | def load_data(file_dir, cat_shape_dict): |
| 29 | with open(file_dir, 'r') as fin: |
| 30 | for l in fin.readlines(): |
| 31 | shape_id, cat = l.rstrip().split() |
| 32 | cat_shape_dict[cat].append(shape_id) |
| 33 | return cat_shape_dict |
| 34 | |
| 35 | class OurDataset(data.Dataset): |
| 36 |
nothing calls this directly
no outgoing calls
no test coverage detected