Sfoglia il codice sorgente

added capture_device_index support

subDesTagesMitExtraKaese 3 anni fa
parent
commit
02a0fcb23d
1 ha cambiato i file con 6 aggiunte e 1 eliminazioni
  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: