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

Compare commits

base: Yun:aec2b9c95a30af354b366297732fdc7372fd7170
Branches Tags
Yun:master
pradana.aumars:master
..
compare: Yun:25765b8b6f516837e361d29fc7635ebe3e8ff224
Branches Tags
Yun:master
pradana.aumars:master

3 Commits
aec2b9c95a ... 25765b8b6f

Author SHA1 Message Date
Pradana AUMARS
25765b8b6f Import numpy in video.py 2021-07-13 16:39:03 +02:00
Pradana AUMARS
09f5098e19 Fix missing parenthesis 2021-07-13 16:36:35 +02:00
Pradana AUMARS
b005e36fcd Merge pull request 'Add additional image processing' (#1) from Yun/videocr:master into master 
Reviewed-on: pradana.aumars/videocr#1
 2021-07-13 16:16:53 +02:00
1 changed files with 2 additions and 1 deletions
Show Stats Expand all files Collapse all files

3
videocr/video.py
View File

@ -4,6 +4,7 @@ import sys
import multiprocessing import multiprocessing
import pytesseract import pytesseract
import cv2 import cv2
import numpy as np
from . import constants from . import constants
from . import utils from . import utils
@ -69,7 +70,7 @@ class Video:
color_mask = cv2.inRange(hsv, (0, 0, 190), (179, 20, 255)) color_mask = cv2.inRange(hsv, (0, 0, 190), (179, 20, 255))
# apply mask, inverse image so it's black text on white background, add borders to top and bottom # apply mask, inverse image so it's black text on white background, add borders to top and bottom
img = cv2.copyMakeBorder(cv2.bitwise_not(cv2.bitwise_and(img, img, mask=color_mask)), 10, 10, 0, 0, cv2.BORDER_CONSTANT, None, (255,255,255) img = cv2.copyMakeBorder(cv2.bitwise_not(cv2.bitwise_and(img, img, mask=color_mask)), 10, 10, 0, 0, cv2.BORDER_CONSTANT, None, (255,255,255))
config = '--tessdata-dir "{}" --psm 7 -c preserve_interword_spaces=1'.format(constants.TESSDATA_DIR) config = '--tessdata-dir "{}" --psm 7 -c preserve_interword_spaces=1'.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)