| 17 | |
| 18 | |
| 19 | class SDFSamples(torch.utils.data.Dataset): |
| 20 | def __init__( |
| 21 | self, |
| 22 | data_source, |
| 23 | split, |
| 24 | subsample |
| 25 | ): |
| 26 | self.subsample = subsample |
| 27 | self.data_source = data_source |
| 28 | self.npyfiles = get_instance_filenames(data_source, split) |
| 29 | |
| 30 | |
| 31 | def __len__(self): |
| 32 | return len(self.npyfiles) |
| 33 | |
| 34 | def __getitem__(self, idx): |
| 35 | filename = os.path.join( |
| 36 | self.data_source, self.npyfiles[idx] |
| 37 | ) |
| 38 | return unpack_sdf_samples(filename, self.subsample), idx, self.npyfiles[idx] |
| 39 | |
| 40 | |
| 41 | class RGBA2SDF(torch.utils.data.Dataset): |
nothing calls this directly
no outgoing calls
no test coverage detected