From 94a29f112d9ceb86bf956ed86319c0c1ac0856ef Mon Sep 17 00:00:00 2001
From: jiangjiajun <jiangjiajun@baidu.com>
Date: Wed, 17 Jun 2020 13:05:23 +0000
Subject: [PATCH] add x2paddle for paddle export to onnx

---
 paddlex/convertor.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/paddlex/convertor.py b/paddlex/convertor.py
index 31b7340..47fc8a8 100644
--- a/paddlex/convertor.py
+++ b/paddlex/convertor.py
@@ -30,6 +30,10 @@ def export_onnx(model_dir, save_dir, fixed_input_shape):
 
 
 def export_onnx_model(model, save_dir):
+    if model.model_type == "detector" or model.__class__.__name__ == "FastSCNN":
+        logging.error(
+            "Only image classifier models and semantic segmentation models(except FastSCNN) are supported to export to ONNX"
+        )
     try:
         import x2paddle
         if x2paddle.__version__ < '0.7.4':
-- 
GitLab