project-KuuKar
/
kuukar-rpi
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

implement kuukar_cv

This commit is contained in:
Siwat Sirichai 2022-12-11 11:18:06 +07:00
parent 3766284cde
commit 52a5f3d8a8
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
kuukar/kuukar.py
View File

@ -5,6 +5,7 @@ import kuukar.kuukar_motion as kuukar_motion
import kuukar.kuukar_lcd as kuukar_lcd import kuukar.kuukar_lcd as kuukar_lcd
import kuukar.kuukar_sensors as kuukar_sensors import kuukar.kuukar_sensors as kuukar_sensors
import kuukar.kuukar_environment as kuukar_environment import kuukar.kuukar_environment as kuukar_environment
import kuukar.kuukar_cv as kuukar_cv
from kuukar.kuukar_config import SERIAL_MCU, SERIAL_DRV from kuukar.kuukar_config import SERIAL_MCU, SERIAL_DRV
@ -23,3 +24,4 @@ class kuukar:
self.lcd, self.leds, self.sensors) self.lcd, self.leds, self.sensors)
self.collision = kuukar_collision.collision( self.collision = kuukar_collision.collision(
self.mcu, self.lcd, self.leds) self.mcu, self.lcd, self.leds)
self.cv = kuukar_cv.cv(self.lcd, self.leds)

3
kuukar/kuukar_cv.py
View File

@ -18,10 +18,13 @@ class cv:
img = self.capture_image() img = self.capture_image()
stop_sign = self.detect_sign(img) stop_sign = self.detect_sign(img)
def get_road_curve(self, frame):
pass
def capture_image(self): def capture_image(self):
ret, frame = self.camera.read() ret, frame = self.camera.read()
return frame return frame
def detect_sign(self, frame): def detect_sign(self, frame):
stopsign_cascade = cv2.CascadeClassifier('./stopsign_good.xml') stopsign_cascade = cv2.CascadeClassifier('./stopsign_good.xml')
image = frame.array image = frame.array