(self, stage=None)
| 325 | self.config = config |
| 326 | |
| 327 | def setup(self, stage=None): |
| 328 | if stage in [None, 'fit']: |
| 329 | self.train_dataset = ColmapIterableDataset(self.config, 'train') |
| 330 | if stage in [None, 'fit', 'validate']: |
| 331 | self.val_dataset = ColmapDataset(self.config, self.config.get('val_split', 'train')) |
| 332 | if stage in [None, 'test']: |
| 333 | # self.test_dataset = ColmapDataset(self.config, self.config.get('test_split', 'test')) |
| 334 | self.test_dataset = ColmapDataset(self.config, 'train') |
| 335 | if stage in [None, 'predict']: |
| 336 | self.predict_dataset = ColmapDataset(self.config, 'train') |
| 337 | |
| 338 | def prepare_data(self): |
| 339 | pass |
no test coverage detected