Echo/fcb_photo_review
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

使用gpu1加载ocr模型

Browse Source
This commit is contained in:
liuyebo
2024-07-10 12:34:59 +08:00
parent 0a7e5b42e8
commit 534bf4b4b9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
photo_mask.py
View File

@@ -20,7 +20,7 @@ from config.ucloud import BUCKET
from models import ZxPhrec, ZxPhhd from models import ZxPhrec, ZxPhhd
from ucloud import ucloud from ucloud import ucloud
OCR = PaddleOCR(use_angle_cls=False, lang="ch", show_log=False) OCR = PaddleOCR(use_angle_cls=False, lang="ch", show_log=False, gpu_id=1)
def open_image(img_path): def open_image(img_path):