From 52a5f3d8a838cce6e32ea23a9ce95cf6103f8459 Mon Sep 17 00:00:00 2001 From: Siwat Sirichai Date: Sun, 11 Dec 2022 11:18:06 +0700 Subject: [PATCH] implement kuukar_cv --- kuukar/kuukar.py | 2 ++ kuukar/kuukar_cv.py | 3 +++ 2 files changed, 5 insertions(+) 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