(parser: ArgumentParser)
| 12 | parser.add_argument('-device', type=str, default='cuda', choices=['cuda', 'cpu']) |
| 13 | |
| 14 | def add_img_args(parser: ArgumentParser): |
| 15 | parser.add_argument('-img_path', type=str, required=True) |
| 16 | parser.add_argument('-av', type=float, default=2.0) |
| 17 | parser.add_argument('-max_dim', type=int, default=2000) |
| 18 | parser.add_argument('-min_divisor', type=int, default=4) |
| 19 | |
| 20 | def get_predict_parser(): |
| 21 | parser = get_base_parser() |
no outgoing calls
no test coverage detected