SengTak commited on
Commit
07b042d
·
1 Parent(s): 2ce4d93

Update pp_ocr.py

Browse files
Files changed (1) hide show
  1. pp_ocr.py +4 -1
pp_ocr.py CHANGED
@@ -11,7 +11,10 @@ pp_ocrv3 = hub.Module(name=module_name)
11
 
12
  def inference_img(img):
13
  with tempfile.TemporaryDirectory() as tempdir_name:
14
- pp_ocrv3.recognize_text(images=[img], use_gpu=False, output_dir=tempdir_name, visualization=True)
 
 
 
15
  result_names = os.listdir(tempdir_name)
16
  result_image = Image.open(os.path.join(tempdir_name, result_names[0]))
17
  return result_image
 
11
 
12
  def inference_img(img):
13
  with tempfile.TemporaryDirectory() as tempdir_name:
14
+ if module_name == "ch_pp-ocrv3":
15
+ pp_ocrv3.recognize_text(images=[img], use_gpu=False, output_dir=tempdir_name, visualization=True)
16
+ elif module_name == "multi_languages_ocr_db_crnn":
17
+ pp_ocrv3.recognize_text(images=[img], output_dir=tempdir_name, visualization=True)
18
  result_names = os.listdir(tempdir_name)
19
  result_image = Image.open(os.path.join(tempdir_name, result_names[0]))
20
  return result_image