| 13 | |
| 14 | |
| 15 | def get_parser(): |
| 16 | parser = argparse.ArgumentParser( |
| 17 | description='Pytorch Referring Expression Segmentation') |
| 18 | parser.add_argument('--config', |
| 19 | default='path to xxx.yaml', |
| 20 | type=str, |
| 21 | help='config file') |
| 22 | parser.add_argument('--opts', |
| 23 | default=None, |
| 24 | nargs=argparse.REMAINDER, |
| 25 | help='override some settings in the config.') |
| 26 | args = parser.parse_args() |
| 27 | assert args.config is not None |
| 28 | cfg = config.load_cfg_from_cfg_file(args.config) |
| 29 | if args.opts is not None: |
| 30 | cfg = config.merge_cfg_from_list(cfg, args.opts) |
| 31 | return cfg |
| 32 | |
| 33 | |
| 34 | def count_parameters(model): |