fix #339

62fc4625 · wenlihaoyu · 7cfa280c · 62fc4625 · 62fc4625
隐藏空白更改
内联并排

Showing with 6 addition and 1 deletion

app.py app.py +4 -1

config.py config.py +2 -0

未找到文件。
--- a/app.py
+++ b/app.py
@@ -144,8 +144,11 @@ class OCR:
                if textLine:
                    ##单行识别
                    partImg = Image.fromarray(img)
-                    text    = ocr.predict(partImg.convert('L'))
+                    text    = crnn.predict(partImg.convert('L'))
                    res =[ {'text':text,'name':'0','box':[0,0,W,0,W,H,0,H]} ]
+                    os.remove(filelock)
+                    break
+                        
                else:
                    detectAngle = textAngle
                    result,angle= model.model(img,

--- a/config.py
+++ b/config.py
@@ -51,3 +51,5 @@ ocrModelTorchEng         = os.path.join(pwd,"models","ocr-english.pth")
 ocrModelOpencv           = os.path.join(pwd,"models","ocr.pb")

 ######################OCR模型###################################################
+
+TIMEOUT=30##超时时间