DustIn-RaspberryPi/image.py at master · faizsiddiqui/DustIn-RaspberryPi · GitHub

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
'''
@task	: 	Computer Vision (Detecting Hands)
@sensor	:	Camera Module
'''

import cv2
import numpy as np
import base64
import time
import urllib2
from Camera import IPCamera as camera

def main():
    cam = camera.IPCamera('http://192.168.43.1:8080/video') #cam = Camera()

    while(cv2.waitKey(1)):
        original = cam.get_frame()
        if original is None:
    		print "error: frame not read from webcam\n"
    		os.system("pause")
    		break
        gray = cv2.cvtColor(original, cv2.COLOR_BGR2GRAY)
        blur = cv2.GaussianBlur(gray, (5, 5), 0)
        frame, threshold = cv2.threshold(blur,70,255,cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)

        image, contours, hierarchy = cv2.findContours(threshold,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)
        drawing = np.zeros(original.shape, np.uint8)

        max_area = 0

        for i in range(len(contours)):
            cnt = contours[i]
            area = cv2.contourArea(cnt)
            if(area>max_area):
                max_area = area
                ci = i

        cnt = contours[ci]
        hull = cv2.convexHull(cnt)
        moments = cv2.moments(cnt)

        if moments['m00']!=0:
            cx = int(moments['m10']/moments['m00']) # cx = M10/M00
            cy = int(moments['m01']/moments['m00']) # cy = M01/M00

        centre = (cx,cy)
        cv2.circle(original, centre, 5, [0,0,255], 2)
        cv2.drawContours(drawing,[cnt],0,(0,255,0),2)
        cv2.drawContours(drawing,[hull],0,(0,0,255),2)

        cnt = cv2.approxPolyDP(cnt,0.01*cv2.arcLength(cnt,True),True)
        hull = cv2.convexHull(cnt,returnPoints = False)

        if(1):
            defects = cv2.convexityDefects(cnt,hull)
            mind=0
            maxd=0
            for i in range(defects.shape[0]):
                s,e,f,d = defects[i,0]
                start = tuple(cnt[s][0])
                end = tuple(cnt[e][0])
                far = tuple(cnt[f][0])
                dist = cv2.pointPolygonTest(cnt,centre,True)
                cv2.line(original,start,end,[0,255,0],2)
                cv2.circle(original,far,5,[0,0,255],-1)

            print(i)
            i=0

        cv2.namedWindow("output", cv2.WINDOW_NORMAL)
        cv2.imshow('output', drawing)  # show windows

        k = cv2.waitKey(10)
        if k == 27:
            break

    cam.get_camera().release()
    cv2.destroyAllWindows()   # remove windows from memory
    return

if __name__ == "__main__":
    main()