Skip to content

Commit 1b4dfa4

Browse files
committed
fixed a cuda config bug
1 parent 6fed221 commit 1b4dfa4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ def __init__(self, gpu=False, times=1, filter=0.8, ua="ARNRS", debug=False):
2323
self.debug = debug
2424

2525
# Init detector and OCR
26-
self.__detector = Detector(device="gpu" if gpu else "cpu")
26+
self.__detector = Detector(device="cuda" if gpu else "cpu")
2727
self.__eocr = easyocr.Reader(['ch_sim', 'en'], gpu=self.gpu)
2828
self.__pocr = PaddleOCR(use_angle_cls=True, use_gpu=self.gpu, show_log=debug)
2929

0 commit comments

Comments
 (0)