| |
| import logging |
|
|
| from .canvas import Canvas |
| from .painters import HAWPainter |
| import matplotlib |
| LOG = logging.getLogger(__name__) |
|
|
| def cli(parser): |
| group = parser.add_argument_group('show') |
|
|
| assert not Canvas.show |
| group.add_argument('--show', default=False,action='store_true', |
| help='show every plot, i.e., call matplotlib show()') |
|
|
| group.add_argument('--edge-threshold', default=None, type=float, |
| help='show the wireframe edges whose confidences are greater than [edge_threshold]') |
| group.add_argument('--out-ext', default='png', type=str, |
| help='save the plot in specific format') |
| def configure(args): |
| Canvas.show = args.show |
| Canvas.out_file_extension = args.out_ext |
| if args.edge_threshold is not None: |
| HAWPainter.confidence_threshold = args.edge_threshold |
