diff --git a/docker-compose.yml b/docker-compose.yml index 96aa3d0..7a21534 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -108,7 +108,7 @@ services: resources: reservations: devices: - - device_ids: [ "0", "1" ] + - device_ids: [ "1" ] capabilities: [ "gpu" ] driver: "nvidia" @@ -126,7 +126,7 @@ services: resources: reservations: devices: - - device_ids: [ "0", "1" ] + - device_ids: [ "1" ] capabilities: [ "gpu" ] driver: "nvidia" @@ -144,6 +144,6 @@ services: resources: reservations: devices: - - device_ids: [ "0", "1" ] + - device_ids: [ "1" ] capabilities: [ "gpu" ] driver: "nvidia" \ No newline at end of file diff --git a/photo_mask/__init__.py b/photo_mask/__init__.py index df55e17..c019dec 100644 --- 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,