| 61 | |
| 62 | |
| 63 | def trigger_visualization_hook(cfg, args): |
| 64 | default_hooks = cfg.default_hooks |
| 65 | if 'visualization' in default_hooks: |
| 66 | visualization_hook = default_hooks['visualization'] |
| 67 | # Turn on visualization |
| 68 | visualization_hook['draw'] = True |
| 69 | if args.show: |
| 70 | visualization_hook['show'] = True |
| 71 | visualization_hook['wait_time'] = args.wait_time |
| 72 | if args.show_dir: |
| 73 | visualization_hook['test_out_dir'] = args.show_dir |
| 74 | visualization_hook['vis_task'] = args.task |
| 75 | else: |
| 76 | raise RuntimeError( |
| 77 | 'VisualizationHook must be included in default_hooks.' |
| 78 | 'refer to usage ' |
| 79 | '"visualization=dict(type=\'VisualizationHook\')"') |
| 80 | |
| 81 | return cfg |
| 82 | |
| 83 | |
| 84 | def main(): |