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

Isolate subtitles as black over white background (kudos to u/Yun on hexbear.net)

Browse Source
This commit is contained in:
Pradana AUMARS 2021-07-12 22:20:00 +02:00
parent 9b97c99657
commit 5534ae317f
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
videocr/video.py
View File

@ -56,6 +56,7 @@ class Video:
if not self.use_fullframe: if not self.use_fullframe:
# 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:, :]
img = cv2.bitwise_not(cv2.bitwise_and(img, img, mask=cv2.inRange(img, (190, 190, 190), (255, 255, 255))))
config = '--tessdata-dir "{}"'.format(constants.TESSDATA_DIR) config = '--tessdata-dir "{}"'.format(constants.TESSDATA_DIR)
try: try:
return pytesseract.image_to_data(img, lang=self.lang, config=config) return pytesseract.image_to_data(img, lang=self.lang, config=config)