(sample_tensor, num_views)
| 34 | |
| 35 | |
| 36 | def reshape_sample_tensor(sample_tensor, num_views): |
| 37 | if num_views == 1: |
| 38 | return sample_tensor |
| 39 | # Need to repeat sample_tensor along the batch dim num_views times |
| 40 | sample_tensor = sample_tensor.unsqueeze(dim=1) |
| 41 | sample_tensor = sample_tensor.repeat(1, num_views, 1, 1) |
| 42 | sample_tensor = sample_tensor.view( |
| 43 | sample_tensor.shape[0] * sample_tensor.shape[1], |
| 44 | sample_tensor.shape[2], |
| 45 | sample_tensor.shape[3] |
| 46 | ) |
| 47 | return sample_tensor |
| 48 | |
| 49 | |
| 50 | def gen_mesh(opt, net, cuda, data, save_path, use_octree=True): |
no outgoing calls
no test coverage detected