hackjam_hackathon/music.py at master · luchris429/hackjam_hackathon · 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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
from scikits import audiolab
import numpy as np
import cv2
import time

sound_dir = "trumpet/"
shortest = 4410

c = audiolab.wavread(sound_dir + "c.wav")[0]
d = audiolab.wavread(sound_dir + "d.wav")[0]
e = audiolab.wavread(sound_dir + "e.wav")[0]
g = audiolab.wavread(sound_dir + "g.wav")[0]
a = audiolab.wavread(sound_dir + "a.wav")[0]

c = c[:shortest]
d = d[:shortest]
e = e[:shortest]
g = g[:shortest]
a = a[:shortest]

cap = cv2.VideoCapture(0)
#cap.set(CV_CAP_PROP_FPS, )

H_MIN = 0
H_MAX = 256
S_MIN = 0
S_MAX = 256
V_MIN = 0
V_MAX = 256
FRAME_WIDTH = 640.0
FRAME_HEIGHT = 480.0
FRAME_SPLIT = FRAME_WIDTH / 5
MAX_NUM_OBJECTS = 10
MIN_OBJECT_AREA = 20*20
MAX_OBJECT_AREA = int(FRAME_HEIGHT * FRAME_WIDTH / 1.5)


def nothing(x):
    pass

cv2.namedWindow('image')
cv2.namedWindow('image2')
cv2.createTrackbar('H_MIN','image',28,255,nothing)
cv2.createTrackbar('S_MIN','image',62,255,nothing)
cv2.createTrackbar('V_MIN','image',11,255,nothing)
cv2.createTrackbar('H_MAX','image',55,255,nothing)
cv2.createTrackbar('S_MAX','image',255,255,nothing)
cv2.createTrackbar('V_MAX','image',174,255,nothing)

switch = '0: OFF \n1 : ON'
cv2.createTrackbar(switch,'image',0,1,nothing)

while(True):
        # Capture frame-by-frame
    ret, frame = cap.read()

    # Our operations on the frame come here
    #blurred = cv2.GaussianBlur(frame, (11, 11), 0)
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)

    # Display the resulting frame
    h_min = cv2.getTrackbarPos('H_MIN', 'image')
    h_max = cv2.getTrackbarPos('H_MAX', 'image')
    s_min = cv2.getTrackbarPos('S_MIN', 'image')
    s_max = cv2.getTrackbarPos('S_MAX', 'image')
    v_min = cv2.getTrackbarPos('V_MIN', 'image')
    v_max = cv2.getTrackbarPos('V_MAX', 'image')


    lower = np.array([h_min, s_min, v_min])
    upper = np.array([h_max, s_max, v_max])
    on = cv2.getTrackbarPos(switch,'image')

    mask = cv2.inRange(hsv, lower, upper)
    mask = cv2.erode(mask, None, iterations=2)
    mask = cv2.dilate(mask, None, iterations=2)

    if cv2.waitKey(1) & 0xFF == ord('q') or on:
        break

    cv2.imshow('image2', mask)
    # When everything done, release the capture
cv2.destroyAllWindows()

while(True):
        # Capture frame-by-frame
    ret, frame = cap.read()

    # Our operations on the frame come here
    #blurred = cv2.GaussianBlur(frame, (11, 11), 0)
    hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)

    mask = cv2.inRange(hsv, lower, upper)
    mask = cv2.erode(mask, None, iterations=2)
    mask = cv2.dilate(mask, None, iterations=2)

    contours = cv2.findContours(mask.copy(), cv2.RETR_EXTERNAL,
                                cv2.CHAIN_APPROX_SIMPLE)[-2]

    if len(contours) > 0:
        c1 = max(contours, key=cv2.contourArea)
        m = cv2.moments(c1)
        center = (int(m['m10'] / m['m00']), int(m['m01'] / m['m00']))
        center_x = center[0]
        if center_x < FRAME_SPLIT:
            audiolab.play(a)
        elif center_x < FRAME_SPLIT * 2:
            audiolab.play(g)
        elif center_x < FRAME_SPLIT * 3:
            audiolab.play(e)
        elif center_x < FRAME_SPLIT * 4:
            audiolab.play(d)
        elif center_x < FRAME_SPLIT * 5:
            audiolab.play(c)
    if cv2.waitKey(1) & 0xFF == ord('q'):
            break
    cv2.imshow('image2', mask)
        # When everything done, release the capture
cap.release()
cv2.destroyAllWindows()