提交 9d8212aa 编写于 作者: S syyxsxx

add yolov3 convertor

上级 553a46a6
......@@ -28,7 +28,7 @@ def arg_parser():
"--data_type",
"-dp",
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
......@@ -61,6 +61,10 @@ def export_openvino_model(model, args):
onnx_parser.set_defaults(output_dir=args.save_dir)
shape = '[1,3,'
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)
mo.main(onnx_parser,'onnx')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册