Adapt to rpi-camera

detect.py

@@ -9,8 +9,15 @@ import numpy as np
 from skimage import io
 import colorsys
 
-img_path = 'IMG_20211204_220355.jpg'
+img_path = 'input.jpg'
 
+from picamera import PiCamera
+camera = PiCamera()
+time.sleep(2) # wait for the camera to initialize
+camera.capture(img_path)
+
+
+# Read the result
 img = io.imread(img_path)[:, :, :]
 
 # Average color

requirements.txt

@@ -0,0 +1,4 @@
+picamera
+opencv-python
+matplotlib
+numpy