Yun/videocr
1
0
Fork 0
forked from pradana.aumars/videocr
Code Issues Pull Requests Projects Releases Wiki Activity

print muted exception from multiprocessing pool

Browse Source
This commit is contained in:
Yi Ge 2019-12-15 22:29:13 +08:00
parent 7b607e5ca7
commit 8f8f2d6d79
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
videocr/video.py
View File

@ -1,4 +1,6 @@
from __future__ import annotations from __future__ import annotations
from typing import List
import sys
import multiprocessing import multiprocessing
import pytesseract import pytesseract
import cv2 import cv2
@ -55,7 +57,10 @@ class Video:
# only use bottom half of the frame by default # only use bottom half of the frame by default
img = img[self.height // 2:, :] img = img[self.height // 2:, :]
config = '--tessdata-dir "{}"'.format(constants.TESSDATA_DIR) config = '--tessdata-dir "{}"'.format(constants.TESSDATA_DIR)
return pytesseract.image_to_data(img, lang=self.lang, config=config) try:
return pytesseract.image_to_data(img, lang=self.lang, config=config)
except Exception as e:
sys.exit('{}: {}'.format(e.__class__.__name__, e))
def get_subtitles(self, sim_threshold: int) -> str: def get_subtitles(self, sim_threshold: int) -> str:
self._generate_subtitles(sim_threshold) self._generate_subtitles(sim_threshold)