(self, args='')
| 391 | return opt |
| 392 | |
| 393 | def init(self, args=''): |
| 394 | # only used in demo |
| 395 | default_dataset_info = { |
| 396 | 'ctdet': 'coco', 'multi_pose': 'coco_hp', 'ddd': 'nuscenes', |
| 397 | 'tracking,ctdet': 'coco', 'tracking,multi_pose': 'coco_hp', |
| 398 | 'tracking,ddd': 'nuscenes' |
| 399 | } |
| 400 | opt = self.parse() |
| 401 | from dataset.dataset_factory import dataset_factory |
| 402 | train_dataset = default_dataset_info[opt.task] \ |
| 403 | if opt.task in default_dataset_info else 'coco' |
| 404 | dataset = dataset_factory[train_dataset] |
| 405 | opt = self.update_dataset_info_and_set_heads(opt, dataset) |
| 406 | return opt |
nothing calls this directly
no test coverage detected