From 1f188ffcec560e97ba22e043c7968427f13e1dd9 Mon Sep 17 00:00:00 2001
From: Channingss <chen_lingchi@163.com>
Date: Wed, 26 Aug 2020 09:00:34 +0000
Subject: [PATCH] optimize code

---
 paddlex/convertor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/paddlex/convertor.py b/paddlex/convertor.py
index 937c42c..9aafa8d 100644
--- a/paddlex/convertor.py
+++ b/paddlex/convertor.py
@@ -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"
-- 
GitLab