Explorar el Código

改为gpu测试

root hace 1 año
padre
commit
615c55d6b3
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  1. 1 1
      deploy/docker_compose/server.py

+ 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_gpu=False, use_angle_cls=True, lang="ch", rec=True)
+ocr = PaddleOCR(use_gpu=True, use_angle_cls=True, lang="ch", rec=True)
 
 def debugg_log(content: str):
     with open("log.txt","a") as file: