Ghink/arnrs
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed a cuda config bug

Browse Source
This commit is contained in:
Bigsk 2023-01-27 13:41:39 +08:00
parent 6fed221f74
commit 1b4dfa4449
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@ -23,7 +23,7 @@ class number(object):
self.debug = debug self.debug = debug
# Init detector and OCR # Init detector and OCR
self.__detector = Detector(device="gpu" if gpu else "cpu") self.__detector = Detector(device="cuda" if gpu else "cpu")
self.__eocr = easyocr.Reader(['ch_sim', 'en'], gpu=self.gpu) self.__eocr = easyocr.Reader(['ch_sim', 'en'], gpu=self.gpu)
self.__pocr = PaddleOCR(use_angle_cls=True, use_gpu=self.gpu, show_log=debug) self.__pocr = PaddleOCR(use_angle_cls=True, use_gpu=self.gpu, show_log=debug)