Remove misnamed module
parent
96497d7327
commit
f77425fd9e
@ -1,30 +0,0 @@
|
||||
import sys
|
||||
|
||||
import cv2
|
||||
import pytesseract
|
||||
|
||||
def crop_to_text(image):
|
||||
kernel = cv2.getStructuringElement(cv2.MORPH_CROSS, (4, 4))
|
||||
opened = cv2.morphologyEx(~image, cv2.MORPH_OPEN, kernel)
|
||||
|
||||
contours, hierarchy = cv2.findContours(opened, cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)
|
||||
bounding_rects = [cv2.boundingRect(c) for c in contours]
|
||||
# The largest contour is certainly the text that we're looking for.
|
||||
largest_rect = max(bounding_rects, key=lambda r: r[2] * r[3])
|
||||
x, y, w, h = largest_rect
|
||||
cropped = image[y:y+h, x:x+w]
|
||||
bordered = cv2.copyMakeBorder(cropped, 5, 5, 5, 5, cv2.BORDER_CONSTANT, None, 255)
|
||||
return bordered
|
||||
def ocr_image(image, config):
|
||||
cropped = crop_to_text(image)
|
||||
return pytesseract.image_to_string(
|
||||
~cropped,
|
||||
config=config
|
||||
)
|
||||
|
||||
def main(f):
|
||||
image = cv2.imread(f, cv2.IMREAD_GRAYSCALE)
|
||||
print(ocr_image(image, "--psm 7"))
|
||||
|
||||
if __name__ == "__main__":
|
||||
main(sys.argv[1])
|
Loading…
Reference in New Issue