From 25ff7eb517452016cd0a9db25d5f152ba1dd0986 Mon Sep 17 00:00:00 2001
From: WenmuZhou <zjwenmu@gmail.com>
Date: Wed, 16 Sep 2020 10:43:11 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=96=B9=E5=90=91=E5=88=86?=
 =?UTF-8?q?=E7=B1=BB=E5=99=A8=E5=AF=BC=E5=87=BA=E6=94=AF=E6=8C=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ppocr/modeling/architectures/cls_model.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ppocr/modeling/architectures/cls_model.py b/ppocr/modeling/architectures/cls_model.py
index 6df20770..ad3ad0e7 100755
--- a/ppocr/modeling/architectures/cls_model.py
+++ b/ppocr/modeling/architectures/cls_model.py
@@ -79,6 +79,7 @@ class ClsModel(object):
             outputs = {'total_loss': loss, 'decoded_out': \
                 predicts['decoded_out'], 'label': label, 'acc': acc}
             return loader, outputs
-
+        elif mode == "export":
+            return [image, predicts]
         else:
             return loader, predicts
-- 
GitLab