| 74 | |
| 75 | |
| 76 | def trigger_visualization_hook(cfg, args): |
| 77 | default_hooks = cfg.default_hooks |
| 78 | if 'visualization' in default_hooks: |
| 79 | visualization_hook = default_hooks['visualization'] |
| 80 | # Turn on visualization |
| 81 | visualization_hook['draw'] = True |
| 82 | if args.show: |
| 83 | visualization_hook['show'] = True |
| 84 | visualization_hook['wait_time'] = args.wait_time |
| 85 | if args.show_dir: |
| 86 | visualization_hook['test_out_dir'] = args.show_dir |
| 87 | all_task_choices = [ |
| 88 | 'mono_det', 'multi-view_det', 'lidar_det', 'lidar_seg', |
| 89 | 'multi-modality_det' |
| 90 | ] |
| 91 | assert args.task in all_task_choices, 'You must set '\ |
| 92 | f"'--task' in {all_task_choices} in the command " \ |
| 93 | 'if you want to use visualization hook' |
| 94 | visualization_hook['vis_task'] = args.task |
| 95 | visualization_hook['score_thr'] = args.score_thr |
| 96 | else: |
| 97 | raise RuntimeError( |
| 98 | 'VisualizationHook must be included in default_hooks.' |
| 99 | 'refer to usage ' |
| 100 | '"visualization=dict(type=\'VisualizationHook\')"') |
| 101 | |
| 102 | return cfg |
| 103 | |
| 104 | |
| 105 | def main(): |