调整显卡调用

2024-08-02 09:54:06 +08:00
parent 45a2289b99
commit 8a56308b03
2 changed files with 4 additions and 4 deletions
--- a/photo_mask/init.py
+++ b/photo_mask/init.py
@@ -39,7 +39,7 @@ SIMILAR_CHAR = {

 # 如果不希望识别出空格，可以设置use_space_char=False。做此项设置一定要测试，2.7.3版本此项设置有bug，会导致识别失败
 OCR = PaddleOCR(
-    gpu_id=1,
+    gpu_id=0,
    show_log=False,
    det_db_thresh=0.1,
    det_db_box_thresh=0.3,