raspberrypi4 LEDの制御

今回は今まで次の表情認識を行うまで

ずっとライトがつきっぱなしであったが

次の表情認識を行うまでライトを消すということを行う。



✳︎プログラム

1. import cognitive_face as CF
2. import json
3. import picamera
4. import time
5. import cv2 as cv
6. import pygame
7. import time
8. import RPi.GPIO as GPIO
9. import time, sys
10. import pygame
11. import time
13.  
14. #ポート番号の指定
15. RED = 17
16. GREEN = 27
17. BLUE = 22
18.  
19. #GPIOポートの初期化
20. GPIO.setmode(GPIO.BCM)
21. ports = [RED, GREEN, BLUE]
22. for port in ports:
23.     GPIO.setup(port, GPIO.OUT)
27. KEY = ''
28. ENDPOINT = ''
30. CF.Key.set(KEY)
31. CF.BaseUrl.set(ENDPOINT)
33. while True:
34.     with picamera.PiCamera() as camera:
35.         camera.resolution = (512,384)
36.         camera.capture('test.jpg')
37.         print('capture')
38.         img = cv.imread('test.jpg')
39.         grayimg = cv.cvtColor(img,cv.COLOR_BGR2GRAY)
40.         img = "test.jpg"
41.         faces = CF.face.detect(img, attributes='emotion')
42.         f=json.dumps(faces)
43.         j=json.loads(f)
45.         for i in range(0,len(j)):
46.             print(j[i]['faceAttributes']['emotion']['happiness'])
47.             happiness_val = j[i]['faceAttributes']['emotion']['happiness']
48.             
49.             print(j[i]['faceAttributes']['emotion']['sadness'])
50.             sadness_val = j[i]['faceAttributes']['emotion']['sadness']
51.             
52.             print(j[i]['faceAttributes']['emotion']['anger'])
53.             anger_val = j[i]['faceAttributes']['emotion']['anger']
54.             
55.             
56.             
57.             
58.             
59.             
60.             if happiness_val > 0.5 :
61.                 
62.                 GPIO.output(RED,0)
63.                 GPIO.output(GREEN,1)
64.                 GPIO.output(BLUE, 0)
65.                 print("幸せ")
66.                 
67.                 
68.                 pygame.mixer.init(frequency=44100)
69.                 pygame.mixer.music.load("/home/pi/mypython/camera/audio/cat1a.mp3")
70.                 pygame.mixer.music.play(1)
71.                 time.sleep(5)
72.                 pygame.mixer.music.stop()
73.                 
74.                 
75.                 GPIO.output(RED,0)
76.                 GPIO.output(GREEN,0)
77.                 GPIO.output(BLUE, 0)
78.                 
79.                 
80.                 
81.             
85.             elif sadness_val > 0.5 :
86.                 
87.                     GPIO.output(RED,0)
88.                     GPIO.output(GREEN,0)
89.                     GPIO.output(BLUE, 1)
90.                     print("悲しい")
91.                 
92.                 
93.                     pygame.mixer.init(frequency=44100)
94.                     pygame.mixer.music.load("/home/pi/mypython/camera/audio/old_cat1.mp3")
95.                     pygame.mixer.music.play(1)
96.                     time.sleep(5)
97.                     pygame.mixer.music.stop()
98.                     
99.                     
100.                     
102.                     GPIO.output(RED,0)
103.                     GPIO.output(GREEN,0)
104.                     GPIO.output(BLUE, 0)
106.             
107.                         
108.             elif anger_val > 0.5 :
109.                 
110.                     GPIO.output(RED,1)
111.                     GPIO.output(GREEN,0)
112.                     GPIO.output(BLUE, 0)
113.                     print("怒ってる")
114.                 
115.                 
116.                     pygame.mixer.init(frequency=44100)
117.                     pygame.mixer.music.load("/home/pi/mypython/camera/audio/purring.mp3")
118.                     pygame.mixer.music.play(1)
119.                     time.sleep(5)
120.                     pygame.mixer.music.stop()
121.                     
122.                     
123.                     GPIO.output(RED,0)
124.                     GPIO.output(GREEN,0)
125.                     GPIO.output(BLUE, 0)