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

optimize code

上级 9b19e8a8
......@@ -31,7 +31,7 @@ def export_onnx(model_dir, save_dir, fixed_input_shape):
def export_onnx_model(model, save_dir, opset_version=10):
if model.__class__.__name__ == "FastSCNN" or (
model.model_type != "detector" and
model.model_type == "detector" and
model.__class__.__name__ != "YOLOv3"):
logging.error(
"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.
先完成此消息的编辑!
想要评论请 注册