forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix export python path

This commit is contained in:
littletomatodonkey 2020-10-27 12:05:30 +00:00
parent db6f83de78
commit 8293b6b07d
3 changed files with 44 additions and 41 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
deploy/hubserving/ocr_det/module.py
View File

@ -3,11 +3,16 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
import os
import sys
__dir__ = os.path.dirname(os.path.abspath(__file__))
sys.path.append(__dir__)
sys.path.append(os.path.abspath(os.path.join(__dir__, '../../../')))
import argparse import argparse
import ast import ast
import copy import copy
import math import math
import os
import time import time
from paddle.fluid.core import AnalysisConfig, create_paddle_predictor, PaddleTensor from paddle.fluid.core import AnalysisConfig, create_paddle_predictor, PaddleTensor
@ -67,9 +72,7 @@ class OCRDet(hub.Module):
images.append(img) images.append(img)
return images return images
def predict(self, def predict(self, images=[], paths=[]):
images=[],
paths=[]):
""" """
Get the text box in the predicted images. Get the text box in the predicted images.
Args: Args:
@ -99,11 +102,9 @@ class OCRDet(hub.Module):
rec_res_final = [] rec_res_final = []
for dno in range(len(dt_boxes)): for dno in range(len(dt_boxes)):
rec_res_final.append( rec_res_final.append({
{
'text_region': dt_boxes[dno].astype(np.int).tolist() 'text_region': dt_boxes[dno].astype(np.int).tolist()
} })
)
all_results.append(rec_res_final) all_results.append(rec_res_final)
return all_results return all_results

18
deploy/hubserving/ocr_rec/module.py
View File

@ -3,11 +3,16 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
import os
import sys
__dir__ = os.path.dirname(os.path.abspath(__file__))
sys.path.append(__dir__)
sys.path.append(os.path.abspath(os.path.join(__dir__, '../../../')))
import argparse import argparse
import ast import ast
import copy import copy
import math import math
import os
import time import time
from paddle.fluid.core import AnalysisConfig, create_paddle_predictor, PaddleTensor from paddle.fluid.core import AnalysisConfig, create_paddle_predictor, PaddleTensor
@ -67,9 +72,7 @@ class OCRRec(hub.Module):
images.append(img) images.append(img)
return images return images
def predict(self, def predict(self, images=[], paths=[]):
images=[],
paths=[]):
""" """
Get the text box in the predicted images. Get the text box in the predicted images.
Args: Args:
@ -99,19 +102,16 @@ class OCRRec(hub.Module):
rec_res, predict_time = self.text_recognizer(img_list) rec_res, predict_time = self.text_recognizer(img_list)
for dno in range(len(rec_res)): for dno in range(len(rec_res)):
text, score = rec_res[dno] text, score = rec_res[dno]
rec_res_final.append( rec_res_final.append({
{
'text': text, 'text': text,
'confidence': float(score), 'confidence': float(score),
} })
)
except Exception as e: except Exception as e:
print(e) print(e)
return [[]] return [[]]
return [rec_res_final] return [rec_res_final]
@serving @serving
def serving_method(self, images, **kwargs): def serving_method(self, images, **kwargs):
""" """

18
deploy/hubserving/ocr_system/module.py
View File

@ -3,11 +3,17 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
import os
import sys
__dir__ = os.path.dirname(os.path.abspath(__file__))
sys.path.append(__dir__)
sys.path.append(os.path.abspath(os.path.join(__dir__, '../../../')))
import argparse import argparse
import ast import ast
import copy import copy
import math import math
import os
import time import time
from paddle.fluid.core import AnalysisConfig, create_paddle_predictor, PaddleTensor from paddle.fluid.core import AnalysisConfig, create_paddle_predictor, PaddleTensor
@ -67,9 +73,7 @@ class OCRSystem(hub.Module):
images.append(img) images.append(img)
return images return images
def predict(self, def predict(self, images=[], paths=[]):
images=[],
paths=[]):
""" """
Get the chinese texts in the predicted images. Get the chinese texts in the predicted images.
Args: Args:
@ -104,13 +108,11 @@ class OCRSystem(hub.Module):
for dno in range(dt_num): for dno in range(dt_num):
text, score = rec_res[dno] text, score = rec_res[dno]
rec_res_final.append( rec_res_final.append({
{
'text': text, 'text': text,
'confidence': float(score), 'confidence': float(score),
'text_region': dt_boxes[dno].astype(np.int).tolist() 'text_region': dt_boxes[dno].astype(np.int).tolist()
} })
)
all_results.append(rec_res_final) all_results.append(rec_res_final)
return all_results return all_results