def get_argument_parser(): parser = get_common_argument_parser() parser.add_argument( '--basenet', default='', help='pretrained base model, should be located in save_folder') parser.add_argument('--test-interval', default=5000, type=int, help='test interval') parser.add_argument("--dataset", help="Dataset to use.", choices=["voc", "coco"], default=None) parser.add_argument('--train_imgs', help='path to training images or VOC root directory') parser.add_argument( '--train_anno', help='path to training annotations or VOC root directory') parser.add_argument('--test_imgs', help='path to testing images or VOC root directory') parser.add_argument( '--test_anno', help='path to testing annotations or VOC root directory') return parser
def get_arguments_parser(): parser = get_common_argument_parser() parser.add_argument("--dataset", help="Dataset to use.", choices=["camvid", "cityscapes", "mapillary"], default=None) return parser
def get_argument_parser(): parser = get_common_argument_parser() parser.add_argument( '--test-every-n-epochs', default=1, type=int, help='Enables running validation every given number of epochs') return parser
def get_argument_parser(): parser = get_common_argument_parser() parser.add_argument( "--dataset", help="Dataset to use.", choices=["imagenet", "cifar100", "cifar10"], default=None ) parser.add_argument('--test-every-n-epochs', default=1, type=int, help='Enables running validation every given number of epochs') return parser
def get_argument_parser(): parser = get_common_argument_parser() parser.add_argument( "--dataset", help="Dataset to use.", choices=["imagenet", "cifar100", "cifar10"], default=None ) parser.add_argument('--test-every-n-epochs', default=1, type=int, help='Enables running validation every given number of epochs') parser.add_argument( "--model_type", choices=[ModelType.KerasLayer, ModelType.FuncModel, ModelType.SubClassModel], default=ModelType.KerasLayer, help="Type of mobilenetV2 model which should be quantized.") return parser
def main(argv): parser = get_common_argument_parser() arguments = parser.parse_args(args=argv) config = create_sample_config(arguments, parser) if arguments.dist_url == "env://": config.update_from_env() if not osp.exists(config.log_dir): os.makedirs(config.log_dir) config.log_dir = str(config.log_dir) configure_paths(config) logger.info("Save directory: {}".format(config.log_dir)) config.execution_mode = get_execution_mode(config) start_worker(main_worker, config)
def main(argv): parser = get_common_argument_parser() arguments = parser.parse_args(args=argv) config = Config.from_json(arguments.config) config.update_from_args(arguments, parser) if config.dist_url == "env://": config.update_from_env() if config.mode.lower() != 'test': if not osp.exists(config.log_dir): os.makedirs(config.log_dir) config.log_dir = str(config.log_dir) configure_paths(config) print("Save directory:", config.log_dir) else: config.log_dir = "/tmp/" config.execution_mode = get_execution_mode(config) start_worker(main_worker, config)