syyxsxx преди 5 години
родител
ревизия
9d8212aae3
променени са 1 файла, в които са добавени 5 реда и са изтрити 1 реда
  1. 5 1
      deploy/openvino/python/convertor.py

+ 5 - 1
deploy/openvino/python/convertor.py

@@ -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')