(parser: ArgumentParser)
| 8 | return parser |
| 9 | |
| 10 | def add_network_args(parser: ArgumentParser): |
| 11 | parser.add_argument('-size', type=str, required=True, choices=['small', 'large', 'defocus_deblur']) |
| 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) |
no outgoing calls
no test coverage detected