“25985edcedea6396277003854657b5f3cb31a628”上不存在“Documentation/filesystems/cifs/AUTHORS”
提交 9d8212aa 编写于 作者: S syyxsxx

add yolov3 convertor

上级 553a46a6
...@@ -28,7 +28,7 @@ def arg_parser(): ...@@ -28,7 +28,7 @@ def arg_parser():
"--data_type", "--data_type",
"-dp", "-dp",
default="FP32", default="FP32",
help="FP32 or FP16, the data_type of openvino IR") help="option, FP32 or FP16, the data_type of openvino IR")
return parser return parser
...@@ -61,6 +61,10 @@ def export_openvino_model(model, args): ...@@ -61,6 +61,10 @@ def export_openvino_model(model, args):
onnx_parser.set_defaults(output_dir=args.save_dir) onnx_parser.set_defaults(output_dir=args.save_dir)
shape = '[1,3,' shape = '[1,3,'
shape = shape + args.fixed_input_shape[1:] shape = shape + args.fixed_input_shape[1:]
if model.__class__.__name__ == "YOLOV3":
shape = shape + ",[1,2]"
inputs = "image,im_size"
onnx_parser.set_defaults(input = inputs)
onnx_parser.set_defaults(input_shape = shape) onnx_parser.set_defaults(input_shape = shape)
mo.main(onnx_parser,'onnx') mo.main(onnx_parser,'onnx')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册