def get_args():
import argparse
import args as A
parser = argparse.ArgumentParser(
description='Inference.')
A.add_runtime_args(parser)
A.add_arch_args(parser)
A.add_val_dataset_args(parser)
A.add_dali_args(parser)
parser.add_argument('--batch-size', '-b', default=100,
type=int, help='Batch size per GPU.')
parser.add_argument('--monitor-path', '-M', default='tmp.val',
help='A directory the results are produced.')
parser.add_argument("weights", help='Path to a trained parameter h5 file.')
parser.add_argument(
'--labels', help='Path to a label name file which contain label names as csv compatible with `label_words.csv`.', default='./label_words.csv')
parser.add_argument(
'--norm-config', '-n', type=A.lower_str, default='default',
help='Specify how to normalize an image as preprocessing.')
parser.add_argument(
'--disable-dataset-size-error', '-E', action='store_false', default=True, dest='raise_dataset_size',
help='If not set, raising an error when the dataset size is divisible by (batch_size * num workers)')
args = parser.parse_args()
# See available archs
A.check_arch_or_die(args.arch)
return args
def get_args():
import argparse
import args as A
parser = argparse.ArgumentParser(description='Inference.')
A.add_runtime_args(parser)
A.add_arch_args(parser)
parser.add_argument("input_image", help='Path to an input image.')
parser.add_argument("weights", help='Path to a trained parameter h5 file.')
parser.add_argument(
'--labels',
help=
'Path to a label name file which contain label names as csv compatible with `label_words.csv`.',
default='./label_words.csv')
parser.add_argument(
'--norm-config',
'-n',
type=A.lower_str,
default='default',
help='Specify how to normalize an image as preprocessing.')
parser.add_argument("--channel-last",
action='store_true',
help='Use a model with NHWC layout.')
parser.add_argument("--spatial-size",
type=int,
default=224,
nargs="+",
help='Spatial size.')
args = parser.parse_args()
# Post process
A.post_process_spatial_size(args)
# See available archs
A.check_arch_or_die(args.arch)
return args
def get_args():
import argparse
import args as A
parser = argparse.ArgumentParser(description='Inference.')
A.add_runtime_args(parser)
A.add_arch_args(parser)
parser.add_argument("input_image", help='Path to an input image.')
parser.add_argument("weights", help='Path to a trained parameter h5 file.')
parser.add_argument(
'--labels',
help=
'Path to a label name file which contain label names as csv compatible with `label_words.csv`.',
default='./label_words.csv')
parser.add_argument(
'--norm-config',
'-n',
type=A.lower_str,
default='default',
help='Specify how to normalize an image as preprocessing.')
args = parser.parse_args()
# See available archs
A.check_arch_or_die(args.arch)
return args