(self, tsdf, name)
| 28 | os.makedirs(self._save_path, exist_ok=True) |
| 29 | |
| 30 | def save_mesh(self, tsdf, name): |
| 31 | if 'semseg' in tsdf.attribute_vols: |
| 32 | mesh = tsdf.get_mesh('semseg') |
| 33 | else: |
| 34 | mesh = tsdf.get_mesh() |
| 35 | |
| 36 | # TODO: include epoch # and train/val |
| 37 | mesh.export(os.path.join(self._save_path, name)) |
| 38 | |
| 39 | |
| 40 | class AtlasLogger(TensorBoardLogger): |
no test coverage detected