调整显存占用

photo_mask/__init__.py

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