-
Notifications
You must be signed in to change notification settings - Fork 1
/
OCRscript.py
62 lines (51 loc) · 1.95 KB
/
OCRscript.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
'''
Script for processing an image, running it through OCR
(Optical Character Recognition), and then writing the results
to a .txt file. Uses OpenCV, PIL, Tesseract, PyTesser.
Part of the Optophone Kit and the Kits for Cultural History
series by the MLab at the University of Victoria (maker.uvic.ca).
Link to the Optophone Kit repository: github.com/uvicmakerlab/optophoneKit
'''
#Import libraries
import cv2
from PIL import Image
from pytesser import *
from picamera.array import PiRGBArray
from picamera import PiCamera
import time
#Save image from webcam
camera = PiCamera()
rawCapture = PiRGBArray(camera)
#allow the camera to warm up
time.sleep(0.1)
#grab an image from the camera
camera.capture(rawCapture, format="bgr")
image = rawCapture.array
cv2.imwrite("test.jpg", image)
'''
To make the image easier for Tesseract to read, convert the image to
grayscale and then use threshold to make the image black and white,
increasing the contrast between text and background.
To read an arbitrary image (rather than one taken with the PiCamera),
you can change the file name below ("imagetoOCR.jpg") to the file
name and/or extension that you want to use and delete the code above
that involves PiCamera.
'''
#Convert image to grayscale
img = cv2.imread("imagetoOCR.jpg") #change file name to match image
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
cv2.imwrite("grayscale.jpg", gray)
#Blur and threshold. Threshold turns the image B&W, increasing the
#contrast between words and background to make words easier to read.
gray = cv2.imread("grayscale.jpg", 0)
blur = cv2.GaussianBlur(gray,(5,5).0)
thresh = cv2.adaptiveThreshold(blur,255,cv2.ADAPTIVE_THRESH_GAUSSIAN_C,\
cv2.THRESH_BINARY,11,2)
cv2.imwrite("threshold.jpg", thresh)
#Finally we get to the OCR
img = Image.open("threshold.jpg")
words = image_to_string(img).strip()
#Write the results to a .txt file
f = open("results.txt", "w")
f.write(words)
f.close()