소스 검색

改为gpu测试

root 1 년 전
부모
커밋
615c55d6b3
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  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: