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)
