修复在训练程序在single模式下，推理结果为None的问题

sml2h3 · Oct 15, 2024 · db75d4a · db75d4a
1 parent 262b017
commit db75d4a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 12 deletions.
diff --git a/README.md b/README.md
@@ -8,8 +8,6 @@ DdddOcr、最简依赖的理念，尽量减少用户的配置和使用成本，
 
 项目地址： [点我传送](https://github.com/sml2h3/ddddocr) 
 
-### 自营GPT聚合平台： [点我传送](https://juxiangyun.com/)
-
 <!-- PROJECT SHIELDS -->
 
 [![Contributors][contributors-shield]][contributors-url]
@@ -22,20 +20,18 @@ DdddOcr、最简依赖的理念，尽量减少用户的配置和使用成本，
 <br />
 
 <p align="center">
-  <a href="https://github.com/shaojintian/Best_README_template/">
+  <a href="https://github.com/sml2h3/ddddocr/">
     <img src="https://cdn.wenanzhe.com/img/logo.png!/crop/700x500a400a500" alt="Logo">
   </a>
   <p align="center">
     一个容易使用的通用验证码识别python库
     <br />
-    <a href="https://github.com/shaojintian/Best_README_template"><strong>探索本项目的文档 »</strong></a>
+    <a href="https://github.com/sml2h3/ddddocr/"><strong>探索本项目的文档 »</strong></a>
     <br />
     <br />
     ·
     <a href="https://github.com/sml2h3/ddddocr/issues">报告Bug</a>
     ·
-    <a href="https://github.com/sml2h3/ddddocr-fastapi">一键API服务</a>
-    ·
     <a href="https://github.com/sml2h3/ddddocr/issues">提出新特性</a>
   </p>
 
@@ -125,10 +121,6 @@ ddddocr
 
 ```
 
-**iii. 一键启动**
-<a href="https://github.com/sml2h3/ddddocr-fastapi">一键API服务</a>
-
-
 ### 项目底层支持 
 
 本项目基于[dddd_trainer](https://github.com/sml2h3/dddd_trainer) 训练所得，训练底层框架位pytorch，ddddocr推理底层抵赖于[onnxruntime](https://pypi.org/project/onnxruntime/)，故本项目的最大兼容性与python版本支持主要取决于[onnxruntime](https://pypi.org/project/onnxruntime/)。

diff --git a/ddddocr/README.md b/ddddocr/README.md
@@ -20,13 +20,13 @@ DdddOcr、最简依赖的理念，尽量减少用户的配置和使用成本，
 <br />
 
 <p align="center">
-  <a href="https://github.com/shaojintian/Best_README_template/">
+  <a href="https://github.com/sml2h3/ddddocr/">
     <img src="https://cdn.wenanzhe.com/img/logo.png!/crop/700x500a400a500" alt="Logo">
   </a>
   <p align="center">
     一个容易使用的通用验证码识别python库
     <br />
-    <a href="https://github.com/shaojintian/Best_README_template"><strong>探索本项目的文档 »</strong></a>
+    <a href="https://github.com/sml2h3/ddddocr/"><strong>探索本项目的文档 »</strong></a>
     <br />
     <br />
     ·

diff --git a/ddddocr/__init__.py b/ddddocr/__init__.py
@@ -2648,6 +2648,7 @@ def classification(self, img, png_fix: bool = False, probability=False):
         if self.__word:
             for item in ort_outs[1]:
                 result.append(self.__charset[item])
+            return ''.join(result)
         else:
             if not self.use_import_onnx:
                 # 概率输出仅限于使用官方模型