()
| 31 | return parser |
| 32 | |
| 33 | def get_ntire_parser(): |
| 34 | parser = get_base_parser() |
| 35 | parser.description=( |
| 36 | "This script produces a submission ready .zip archive to be uploaded at " |
| 37 | "https://www.codabench.org/competitions/12764/#/participate-tab for evaluation by our server. \n" |
| 38 | "Use \'-n [NAME]\' to set name of your architecture in the leaderboard.\n" |
| 39 | "Put the development (and the final test input= image archive to the \'./dataset\' folder.\n" |
| 40 | "When the test phase starts, use \'-p test\' to load the test set.") |
| 41 | parser.add_argument('-name', '-n', type=str, required=True, |
| 42 | help='name of your method to be shown on the leaderboard') |
| 43 | parser.add_argument('-phase', '-p', type=str, choices=['dev', 'test'], default='dev', |
| 44 | help='current phase of the challenge') |
| 45 | parser.add_argument('-checkpoint', '-c', type=str, required=True, |
| 46 | help='name of your checkpoint in the \'./checkpoint\' folder\'.') |
| 47 | parser.add_argument('--extra_data', '--ed', action='store_true', |
| 48 | help='activate if data other than RealBokeh was used to train your model') |
| 49 | parser.add_argument('-device', type=str, default='cuda', choices=['cuda', 'cpu'], |
| 50 | help='device to use') |
| 51 | parser.add_argument('-dataset_root_dir', '-dr', type=Path, default='./dataset', |
| 52 | help='path to the dir containing the Bokeh_NTIRE dataset folder/archive, default is \'./dataset\'') |
| 53 | return parser |
no test coverage detected