소스 검색

add yolov3 convertor

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