test_ocr.py 518 B

123456789101112131415161718192021
  1. import cv2
  2. import time
  3. from onnxocr.onnx_paddleocr import ONNXPaddleOcr,sav2Img
  4. import sys
  5. import time
  6. #固定到onnx路径·
  7. # sys.path.append('./paddle_to_onnx/onnx')
  8. model = ONNXPaddleOcr(use_angle_cls=True, use_gpu=False)
  9. img = cv2.imread('./onnxocr/test_images/715873facf064583b44ef28295126fa7.jpg')
  10. s = time.time()
  11. result = model.ocr(img)
  12. e = time.time()
  13. print("total time: {:.3f}".format(e - s))
  14. print("result:", result)
  15. for box in result[0]:
  16. print(box)
  17. sav2Img(img, result,name=str(time.time())+'.jpg')