from shared pc

Author: daan

flir/flir_capture_single.py

@@ -0,0 +1,120 @@
+import sys
+import time
+import PySpin
+import os
+
+import numpy as np
+import cv2
+
+
+import argparse
+
+# construct the argument parse and parse the arguments
+ap = argparse.ArgumentParser()
+ap.add_argument("-c", "--camera", type=str, default="0", help="camera by id")
+
+args = vars(ap.parse_args())
+
+camera_serial = args["camera"]
+
+
+
+
+def set_trigger_mode_software(cam):
+    cam.TriggerMode.SetValue(PySpin.TriggerMode_Off)
+    cam.TriggerSource.SetValue(PySpin.TriggerSource_Software)
+    cam.TriggerMode.SetValue(PySpin.TriggerMode_On)
+    print("set trigger mode software")
+
+
+def reset_trigger_mode_software(cam):
+    cam.TriggerMode.SetValue(PySpin.TriggerMode_Off)
+    print("reset trigger mode")
+
+
+#
+#   setup
+#
+
+system = PySpin.System.GetInstance()
+cam_list = system.GetCameras()
+
+if cam_list.GetSize() == 0:
+    print("no cameras found, aborting")
+    system.ReleaseInstance()
+    del system
+    sys.exit()
+
+cameras = []
+for i in range(cam_list.GetSize()):
+    cam = cam_list.GetByIndex(i)
+    print("camera {} serial: {}".format(i, cam.GetUniqueID()))
+
+
+#camera_serial = "18284509"
+
+cam = cam_list.GetBySerial(camera_serial)
+
+try:
+    cam.Init()
+    cam.AcquisitionMode.SetValue(PySpin.AcquisitionMode_Continuous)
+    set_trigger_mode_software(cam)
+    cam.BeginAcquisition()
+
+except:
+    print("error initializing camera {}".format(camera_serial))
+    sys.exit()
+
+
+os.mkdir(camera_serial)
+
+
+
+#
+#   loop
+#
+
+count = 0
+
+while 1:
+    key = cv2.waitKey(1)
+
+
+
+    if key == 27: # ESC
+        cv2.destroyAllWindows()
+        break
+    
+    cam.TriggerSoftware()
+    i = cam.GetNextImage()
+        #print(i.GetWidth(), i.GetHeight(), i.GetBitsPerPixel())
+    cvi = None 
+
+    if i.IsIncomplete():
+        pass
+    else:            
+        # see documentation: enum ColorProcessingAlgorithm 
+        image_converted = i.Convert(PySpin.PixelFormat_BGR8, PySpin.DIRECTIONAL_FILTER)
+        image_data = image_converted.GetData()
+        cvi = np.frombuffer(image_data, dtype=np.uint8)            
+        cvi = cvi.reshape((i.GetHeight(),i.GetWidth(),3)) 
+        cv2.imshow("cam1",cvi)
+    i.Release()
+    del i
+
+    if key == 32:
+        cv2.imwrite("{}/capture_{}.jpg".format(camera_serial,count), cvi)
+        count+= 1
+        print("saved image {}".format(count))
+#
+#   cleanup
+#
+
+cam.EndAcquisition()
+reset_trigger_mode_software(cam)
+cam.DeInit()
+del cam
+del cam_list
+
+system.ReleaseInstance()
+del system