|
@@ -21,7 +21,7 @@ if __name__ == '__main__':
|
|
parser.add_argument('--model_filename', default="origin_models", type=str, help='模型文件路径')
|
|
parser.add_argument('--model_filename', default="origin_models", type=str, help='模型文件路径')
|
|
parser.add_argument('--framework', default=None, type=str, help='框架类型 (pytorch 或 tensorflow)')
|
|
parser.add_argument('--framework', default=None, type=str, help='框架类型 (pytorch 或 tensorflow)')
|
|
parser.add_argument('--mode', default=None, type=str, help='验证模式 (blackbox 或 whitebox)')
|
|
parser.add_argument('--mode', default=None, type=str, help='验证模式 (blackbox 或 whitebox)')
|
|
- parser.add_argument('--model_type', default=None, type=str, help='模型名称,可选参数:alexnet、googlenet、resnet、vgg16、ssd、yolox、rcnn')
|
|
|
|
|
|
+ parser.add_argument('--model_type', default=None, type=str, help='模型名称,可选参数:alexnet、googlenet、resnet、vgg16、ssd、yolox、faster_rcnn')
|
|
args, _ = parser.parse_known_args()
|
|
args, _ = parser.parse_known_args()
|
|
result = verify_model(args.model_filename, args.framework, args.mode, args.model_type)
|
|
result = verify_model(args.model_filename, args.framework, args.mode, args.model_type)
|
|
print(f"verify_result: {result}")
|
|
print(f"verify_result: {result}")
|