Explorar o código

added capture_device_index support

subDesTagesMitExtraKaese %!s(int64=3) %!d(string=hai) anos
pai
achega
02a0fcb23d
Modificáronse 1 ficheiros con 6 adicións e 1 borrados
  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: