@@ -52,8 +52,6 @@ add_arg('use_se', bool, True, "Whether to use Squeeze-
defeval(args):
image_shape=args.image_shape
model_list=[mformindir(models)if"__"notinm]
assertargs.modelinmodel_list,"{} is not in lists: {}".format(args.model,
model_list)
...
...
@@ -62,8 +60,11 @@ def eval(args):
),"{} doesn't exist, please load right pretrained model path for eval".format(
args.pretrained_model)
assertargs.image_shape[
1]<=args.resize_short_size,"Please check the args:image_shape and args:resize_short_size, The croped size(image_shape[1]) must smaller than or equal to the resized length(resize_short_size) "
@@ -54,14 +54,17 @@ add_arg('use_se', bool, True, "Whether to use Squeeze-
definfer(args):
image_shape=args.image_shape
model_list=[mformindir(models)if"__"notinm]
assertargs.modelinmodel_list,"{} is not in lists: {}".format(args.model,
model_list)
assertos.path.isdir(args.pretrained_model
),"please load right pretrained model path for infer"
assertargs.image_shape[
1]<=args.resize_short_size,"Please check the args:image_shape and args:resize_short_size, The croped size(image_shape[1]) must smaller than or equal to the resized length(resize_short_size) "