Returns a dictionary that can be saved or load.
(self)
| 2447 | return ray_uv_dict |
| 2448 | |
| 2449 | def state_dict(self) -> T.Dict[str, T.Any]: |
| 2450 | """Returns a dictionary that can be saved or load.""" |
| 2451 | to_save = dict() |
| 2452 | for name in ['rgb', 'depth', 'normal_w', 'hit_map', 'feature']: |
| 2453 | to_save[name] = getattr(self, name, None) |
| 2454 | to_save['camera'] = self.camera.state_dict() |
| 2455 | return to_save |
| 2456 | |
| 2457 | def load_state_dict( |
| 2458 | self, |