-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathz-bluePenAsPointer.py
99 lines (78 loc) · 3.13 KB
/
z-bluePenAsPointer.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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
import requests
import cv2
import numpy as np
import urllib.request # for reading image from URL
import imutils
from PIL import Image
import pytesseract
while True:
# *************************IP webcam image stream ************************************
URL = "http://192.168.43.1:8080/shot.jpg"
urllib.request.urlretrieve(URL, "shot1.jpg")
frame = cv2.imread("shot1.jpg")
# resize the frame, blur it, and convert it to the HSV
frame = imutils.resize(frame, width=640, height=480)
blurred = cv2.GaussianBlur(frame, (11, 11), 0)
hsv = cv2.cvtColor(blurred, cv2.COLOR_BGR2HSV)
# h, w, channels = np.shape(frame)
# print image properties.
# print("width: " + str(w))
# print("height: " + str(h))
# print("channels: " + str(channels))
# define range of blue color in HSV
lower = np.array([110, 50, 50])
upper = np.array([130, 255, 255])
# lower = np.array([30, 150, 50]) # red color- but not accurate
# upper = np.array([255, 255, 180])
# Threshold the HSV image to get only blue colors
mask = cv2.inRange(hsv, lower, upper)
# Bitwise-AND mask and original image # To show filtered images.
# res = cv2.bitwise_and(frame, frame, mask=mask)
# find contours and draw border and calculate points
cnts = cv2.findContours(mask.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)[-2]
center = None
# only proceed if at least one contour was found
if len(cnts) > 0:
# find the largest contour in the mask, then use
# it to compute the minimum enclosing circle and
# centroid
c = max(cnts, key=cv2.contourArea)
((x, y), radius) = cv2.minEnclosingCircle(c)
# cv2.drawContours(frame, c, -1, (255, 0, 0), 2)
# cv2.circle(frame, (int(x),int(y)), int(radius), (0, 255, 255), 2)
cv2.circle(frame,(int(x),int(y)),3,(255,255,0),3) # this is center
#*************** crop that image **************
x1 = int(x - 30)
x2 = int(x + 30)
y1 = int(y - radius -50)
y2 = int(y - radius -00)
pt1 = (x1,y1)
pt2 = (x2,y2)
cv2.rectangle(frame, pt1, pt2 , (255,255,0),3)
# print("x =" +str(x))
# print("y =" +str(y))
# print("radius =" +str(radius))
crop_img = frame[y1:y2, x1:x2]
cv2.imshow("cropped", crop_img)
# ************** Tesseract-part ****************************
im = Image.fromarray(crop_img)
text = pytesseract.image_to_string(im)
print("text-output is = " + text)
# ************** arguments for writting on video-frame *********
font = cv2.FONT_HERSHEY_SIMPLEX
bottomLeftCornerOfText = (10, 50)
fontScale = 1
fontColor = (255, 0, 255)
lineType = 2
text = "CV-KeyBoard"
cv2.putText(
frame, text, bottomLeftCornerOfText, font, fontScale, fontColor, lineType
)
cv2.imshow("Frame", frame)
# cv2.imshow("mask", mask)
# cv2.imshow("res", res)
# **************** boilerplate-code => same for all ************************#
if cv2.waitKey(1) == 27:
break
cap.release()
cv2.destroyAllWindows()