diff --git a/kuukar/kuukar.py b/kuukar/kuukar.py index cf294c7..4c622a5 100644 --- a/kuukar/kuukar.py +++ b/kuukar/kuukar.py @@ -5,6 +5,7 @@ import kuukar.kuukar_motion as kuukar_motion import kuukar.kuukar_lcd as kuukar_lcd import kuukar.kuukar_sensors as kuukar_sensors import kuukar.kuukar_environment as kuukar_environment +import kuukar.kuukar_cv as kuukar_cv from kuukar.kuukar_config import SERIAL_MCU, SERIAL_DRV @@ -23,3 +24,4 @@ class kuukar: self.lcd, self.leds, self.sensors) self.collision = kuukar_collision.collision( self.mcu, self.lcd, self.leds) + self.cv = kuukar_cv.cv(self.lcd, self.leds) diff --git a/kuukar/kuukar_cv.py b/kuukar/kuukar_cv.py index 3705b55..fc2e8cc 100644 --- a/kuukar/kuukar_cv.py +++ b/kuukar/kuukar_cv.py @@ -18,10 +18,13 @@ class cv: img = self.capture_image() stop_sign = self.detect_sign(img) + def get_road_curve(self, frame): + pass def capture_image(self): ret, frame = self.camera.read() return frame + def detect_sign(self, frame): stopsign_cascade = cv2.CascadeClassifier('./stopsign_good.xml') image = frame.array