提交 fa8a0e38 编写于 作者: W WenmuZhou

Add hint

上级 3ce97f18
...@@ -282,8 +282,13 @@ class PaddleOCR(predict_system.TextSystem): ...@@ -282,8 +282,13 @@ class PaddleOCR(predict_system.TextSystem):
if isinstance(img, list) and det == True: if isinstance(img, list) and det == True:
logger.error('When input a list of images, det must be false') logger.error('When input a list of images, det must be false')
exit(0) exit(0)
if cls == False:
self.use_angle_cls = False
elif cls == True and self.use_angle_cls == False:
logger.warning(
'Since the angle classifier is not initialized, the angle classifier will not be uesd during the forward process'
)
self.use_angle_cls = cls
if isinstance(img, str): if isinstance(img, str):
# download net image # download net image
if img.startswith('http'): if img.startswith('http'):
...@@ -301,6 +306,7 @@ class PaddleOCR(predict_system.TextSystem): ...@@ -301,6 +306,7 @@ class PaddleOCR(predict_system.TextSystem):
if isinstance(img, np.ndarray) and len(img.shape) == 2: if isinstance(img, np.ndarray) and len(img.shape) == 2:
img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR) img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
if det and rec: if det and rec:
dt_boxes, rec_res = self.__call__(img) dt_boxes, rec_res = self.__call__(img)
return [[box.tolist(), res] for box, res in zip(dt_boxes, rec_res)] return [[box.tolist(), res] for box, res in zip(dt_boxes, rec_res)]
elif det and not rec: elif det and not rec:
...@@ -342,3 +348,9 @@ def main(): ...@@ -342,3 +348,9 @@ def main():
if result is not None: if result is not None:
for line in result: for line in result:
logger.info(line) logger.info(line)
if __name__ == '__main__':
ocr = PaddleOCR(use_angle_cls=False)
r = ocr.ocr('doc/imgs/1.jpg')
print(r)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册