Parcourir la source

added capture_device_index support

subDesTagesMitExtraKaese il y a 3 ans
Parent
commit
02a0fcb23d
1 fichiers modifiés avec 6 ajouts et 1 suppressions
  1. 6 1
      raspberry-pi/sensors/opticalSensor.py

+ 6 - 1
raspberry-pi/sensors/opticalSensor.py

@@ -110,7 +110,12 @@ class OpticalSensor():
       self._t.start()
 
   def _getFrames(self):
-    self.cap = cv2.VideoCapture(self.conf["opt_sensor"]["capture_device"])
+    try:
+      dev = int(self.conf["opt_sensor"]["capture_device"])
+    except ValueError:
+      dev = self.conf["opt_sensor"]["capture_device"]
+    self.cap = cv2.VideoCapture(dev)
+
     while self.running:
       self.success, image = self.cap.read()
       if self.success: