forked from murtazahassan/Real-Time-Object-Measurement
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathObjectMeasurement.py
47 lines (43 loc) · 1.96 KB
/
ObjectMeasurement.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
import cv2
import utlis
###################################
webcam = True
path = '1.jpg'
cap = cv2.VideoCapture(0)
cap.set(10,160)
cap.set(3,1920)
cap.set(4,1080)
scale = 3
wP = 210 *scale
hP= 297 *scale
###################################
while True:
if webcam:success,img = cap.read()
else: img = cv2.imread(path)
imgContours , conts = utlis.getContours(img,minArea=50000,filter=4)
if len(conts) != 0:
biggest = conts[0][2]
#print(biggest)
imgWarp = utlis.warpImg(img, biggest, wP,hP)
imgContours2, conts2 = utlis.getContours(imgWarp,
minArea=2000, filter=4,
cThr=[50,50],draw = False)
if len(conts) != 0:
for obj in conts2:
cv2.polylines(imgContours2,[obj[2]],True,(0,255,0),2)
nPoints = utlis.reorder(obj[2])
nW = round((utlis.findDis(nPoints[0][0]//scale,nPoints[1][0]//scale)/10),1)
nH = round((utlis.findDis(nPoints[0][0]//scale,nPoints[2][0]//scale)/10),1)
cv2.arrowedLine(imgContours2, (nPoints[0][0][0], nPoints[0][0][1]), (nPoints[1][0][0], nPoints[1][0][1]),
(255, 0, 255), 3, 8, 0, 0.05)
cv2.arrowedLine(imgContours2, (nPoints[0][0][0], nPoints[0][0][1]), (nPoints[2][0][0], nPoints[2][0][1]),
(255, 0, 255), 3, 8, 0, 0.05)
x, y, w, h = obj[3]
cv2.putText(imgContours2, '{}cm'.format(nW), (x + 30, y - 10), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1.5,
(255, 0, 255), 2)
cv2.putText(imgContours2, '{}cm'.format(nH), (x - 70, y + h // 2), cv2.FONT_HERSHEY_COMPLEX_SMALL, 1.5,
(255, 0, 255), 2)
cv2.imshow('A4', imgContours2)
img = cv2.resize(img,(0,0),None,0.5,0.5)
cv2.imshow('Original',img)
cv2.waitKey(1)