提交 1f188ffc 编写于 作者: C Channingss

optimize code

上级 9b19e8a8
...@@ -31,7 +31,7 @@ def export_onnx(model_dir, save_dir, fixed_input_shape): ...@@ -31,7 +31,7 @@ def export_onnx(model_dir, save_dir, fixed_input_shape):
def export_onnx_model(model, save_dir, opset_version=10): def export_onnx_model(model, save_dir, opset_version=10):
if model.__class__.__name__ == "FastSCNN" or ( if model.__class__.__name__ == "FastSCNN" or (
model.model_type != "detector" and model.model_type == "detector" and
model.__class__.__name__ != "YOLOv3"): model.__class__.__name__ != "YOLOv3"):
logging.error( logging.error(
"Only image classifier models, detection models(YOLOv3) and semantic segmentation models(except FastSCNN) are supported to export to ONNX" "Only image classifier models, detection models(YOLOv3) and semantic segmentation models(except FastSCNN) are supported to export to ONNX"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册