This commit is contained in:
WenmuZhou 2021-02-24 15:38:54 +08:00
parent 3ce97f18ec
commit fa8a0e3805
1 changed files with 13 additions and 1 deletions

View File

@ -282,8 +282,13 @@ class PaddleOCR(predict_system.TextSystem):
if isinstance(img, list) and det == True:
logger.error('When input a list of images, det must be false')
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):
# download net image
if img.startswith('http'):
@ -301,6 +306,7 @@ class PaddleOCR(predict_system.TextSystem):
if isinstance(img, np.ndarray) and len(img.shape) == 2:
img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
if det and rec:
dt_boxes, rec_res = self.__call__(img)
return [[box.tolist(), res] for box, res in zip(dt_boxes, rec_res)]
elif det and not rec:
@ -342,3 +348,9 @@ def main():
if result is not None:
for line in result:
logger.info(line)
if __name__ == '__main__':
ocr = PaddleOCR(use_angle_cls=False)
r = ocr.ocr('doc/imgs/1.jpg')
print(r)