Browse Source

修改测试为gpu测试

hans 1 year ago
parent
commit
da5d0bfdc1
2 changed files with 2 additions and 1 deletions
  1. 1 0
      .gitignore
  2. 1 1
      deploy/docker_compose/server.py

+ 1 - 0
.gitignore

@@ -30,5 +30,6 @@ paddleocr.egg-info/
 /deploy/android_demo/app/PaddleLite/
 /deploy/android_demo/app/.cxx/
 /deploy/android_demo/app/cache/
+deploy/docker_compose/venv/
 test_tipc/web/models/
 test_tipc/web/node_modules/

+ 1 - 1
deploy/docker_compose/server.py

@@ -20,7 +20,7 @@ img_dir = Path('imgs')
 img_dir.mkdir(exist_ok=True)
 ocr_cache_dict = dict()
 
-ocr = PaddleOCR(use_angle_cls=True, lang="ch", rec=True)
+ocr = PaddleOCR(use_angle_cls=True, lang="ch", rec=True, use_gpu=True)
 
 def debugg_log(content: str):
     with open("log.txt","a") as file: