| 61 | return parser.parse_known_args()[0] |
| 62 | |
| 63 | def get_life_args(): |
| 64 | parser = configargparse.ArgParser(config_file_parser_class=configargparse.YAMLConfigFileParser) |
| 65 | parser.add_argument('--iters', type=int, default=12) |
| 66 | parser.add_argument('--mixed_precision', type=bool, default=True) |
| 67 | parser.add_argument('--small', action='store_true', help='use small model') |
| 68 | parser.add_argument("--fnet", type=str, default='CNN') |
| 69 | parser.add_argument('--model', type=str, default="./pretrained_models/cnn_one.pth", help="choose the trained model") |
| 70 | args = parser.parse_known_args()[0] |
| 71 | return args |
| 72 | |
| 73 | def get_demo_video_args(): |
| 74 | parser = argparse.ArgumentParser() |