Upload 4 samples file for camera image acquisition

examples/camera/AVTCamera.py

@@ -0,0 +1,89 @@
+from pymba import vimba
+from pymba import Frame
+from time import sleep
+import cv2
+
+PIXEL_FORMATS_CONVERSIONS = {
+    'BayerRG8': cv2.COLOR_BAYER_RG2RGB,
+}
+
+class AVTCamera:
+
+    vimba=None
+    camFactory=None
+
+    def __init__(self,capture_device=0):
+        # vimba object
+        if AVTCamera.vimba == None:
+            AVTCamera.vimba = vimba.Vimba()
+            # Start vimba system
+            AVTCamera.vimba.startup()
+            AVTCamera.camFactory = vimba.camera_ids()
+        self.cam_open=False
+
+        cams = [AVTCamera.vimba.camera(id) for id in AVTCamera.camFactory]
+        if len(cams) == 0 or len(cams) < capture_device:
+            print("No AVT camera found")
+            self.cam=None
+        else:
+            self.cam=cams[capture_device]
+
+    def read(self):
+        self.open()
+        self.singleframe = 0
+        self.cam.start_frame_acquisition()
+        while self.singleframe == 0 :
+            sleep(0.05)
+        self.cam.stop_frame_acquisition()
+        return self.image
+
+    def convert_frame(self, frame: Frame )->None:
+        #print('frame {}'.format(frame.data.frameID))
+        self.singleframe += 1
+        self.image = frame.buffer_data_numpy()
+        try:
+            self.image = cv2.cvtColor(self.image, PIXEL_FORMATS_CONVERSIONS[frame.pixel_format])
+        except KeyError:
+            pass
+
+    def release(self):
+        self.cam.disarm()
+        self.cam.close()
+        self.cam_open = False
+
+    def preview(self):
+        while True:
+            img=self.read()
+            cv2.imshow('xc', img)
+            c=cv2.waitKey(10)
+            if c==ord ('q') :
+                cv2.destroyWindow('xc')
+                return img;
+
+    def open(self):
+        if self.cam == None:
+            self.image=None
+            return self.image
+        if self.cam_open == False :
+            self.cam.open()
+            self.cam_open = True
+            self.cam.arm('Continuous', self.convert_frame)
+
+
+def cam_test():
+    camera=AVTCamera(0)
+    while True:
+        image=camera.read()
+        cv2.imshow('Image', image)
+        c=cv2.waitKey(10)
+        if c==ord ('q') :
+            camera.release()
+            break;
+
+def cam_test_preview():
+    camera=AVTCamera(0)
+    camera.preview()
+    camera.release()
+
+if __name__=='__main__':
+    cam_test_preview()

examples/camera/multicamera_freerun.py

@@ -0,0 +1,59 @@
+from __future__ import print_function
+from typing import Optional
+from time import sleep
+from pymba import Vimba
+from pymba import Frame
+
+vimba = None
+def FrameCallback(frame: Frame, delay: Optional[int] = 10) -> None:
+    bShowFrameInfos = True
+    if bShowFrameInfos == True:
+        print('Frame ID:%d' %(frame.data.frameID), end=" ")
+        print('Size: %d' %(len(frame.buffer_data())), end=" " )
+        print("Format:%s" %(frame.pixel_format))
+        # print('Status: %s' % (frame.receiveStatus)) -- no such feature up to now
+    print(".", end="")
+
+
+def init_cameras():
+    global vimba
+    # vimba object
+    vimba = Vimba()
+    # Start vimba system
+    vimba.startup()
+    vmFactory = vimba.camera_ids()
+    # Get connected cameras
+    cams = [vimba.camera(id) for id in vmFactory]
+    if len(cams) == 0:
+        raise OSError("No camera present.")
+    for idx, device in enumerate(vmFactory):
+        print("Device {} ID: {}".format(idx, device))
+    # Open cameras and print information
+    for idx, cam in enumerate(cams):
+        try:
+            cam.open()
+            cam.arm(mode='Continuous', callback=FrameCallback)
+            cam.start_frame_acquisition()
+        except VimbaException as e:
+            if e.error_code == VimbaException.ERR_TIMEOUT:
+                print(e)
+                cam.disarm()
+                cam.arm('Continuous')
+            elif e.error_code == VimbaException.ERR_DEVICE_NOT_OPENED:
+                print(e)
+                cam.open()
+                cam.arm(mode='Continuous', callback=FrameCallback)
+    return cams
+
+def main():
+    global vimba
+    cams = init_cameras()
+    sleep(2)
+    for cam in cams:
+        cam.disarm()
+        cam.close()
+    vimba.shutdown()
+    print("Exit Program...")
+
+if __name__ == '__main__':
+    main()

examples/camera/opencv_display_various_image.py

@@ -0,0 +1,188 @@
+
+from __future__ import absolute_import, print_function, division
+from pymba import *
+import numpy as np
+import cv2
+import time
+
+Vimba().startup()
+camera = Vimba().camera(0)
+camera.open()
+print("Camera List ID is %s", Vimba.camera_ids())
+
+pixel_format = camera.feature("PixelFormat")
+pixel_format.value = "BayerRG8"
+# Note: Tested with at "Mono8" "BayerRG8" "BayerRG12" "BayerRG12Packed" "RGB8Packed" pixel format
+
+camera.arm('SingleFrame')
+image_Height = camera.feature("Height")
+image_Width = camera.feature("Width")
+Height = image_Height.value
+Width = image_Width.value
+
+# capture a single frame, more than once if desired
+for i in range(1):
+    frame = camera.acquire_frame()
+    print('frame No.{}'.format(frame.data.frameID))
+    camera_frame_size = len(frame.buffer_data())
+    frame_pixel_format = frame.pixel_format
+    print("Frame size: %d,  Image Resolution: %dx%d,  pixel_format: %s " % (camera_frame_size, Width, Height, frame_pixel_format))
+    data_bytes = frame.buffer_data()
+
+    if (frame_pixel_format == "Mono8" or frame_pixel_format == "BayerRG8" or frame_pixel_format == "BayerGR8"):
+        frame_8bits = np.ndarray(buffer=data_bytes, dtype=np.uint8, shape=(Height, Width))
+
+    elif (frame_pixel_format == "BayerRG12" or frame_pixel_format == "Mono10" or frame_pixel_format == "Mono12" or frame_pixel_format == "Mono14"):
+        data_bytes = np.frombuffer(data_bytes, dtype=np.uint8)
+        pixel_even = data_bytes[0::2]
+        pixel_odd = data_bytes[1::2]
+
+        # Convert bayer16 to bayer8 / Convert Mono12/Mono14 to Mono8
+        if (frame_pixel_format == "Mono14"):
+            pixel_even = np.right_shift(pixel_even, 6)
+            pixel_odd = np.left_shift(pixel_odd, 2)
+        elif (frame_pixel_format == "Mono10"):
+            pixel_even = np.right_shift(pixel_even, 2)
+            pixel_odd = np.left_shift(pixel_odd, 6)
+        else:
+            pixel_even = np.right_shift(pixel_even, 4)
+            pixel_odd = np.left_shift(pixel_odd, 4)
+        frame_8bits = np.bitwise_or(pixel_even, pixel_odd).reshape(Height, Width)
+
+    elif (frame_pixel_format == "BayerRG12Packed" or frame_pixel_format == "Mono12Packed" or frame_pixel_format == "BayerGR12Packed"):
+        data_bytes = np.frombuffer(data_bytes, dtype=np.uint8)
+        size = len(data_bytes)
+        index = []
+        for i in range(0, size, 3):
+            index.append(i+1)
+
+        data_bytes = np.delete(data_bytes, index)
+        frame_8bits = data_bytes.reshape(Height, Width)
+
+    elif (frame_pixel_format == "RGB8Packed" or frame_pixel_format == "BGR8Packed"):
+        frame_8bits = np.ndarray(buffer=frame.buffer_data(), dtype=np.uint8, shape=(Height, Width*3))
+
+    else:
+        # Note: wait to do -- other format, such as YUV411Packed, YUV422Packed, YUV444Packed
+        frame_8bits = np.ndarray(buffer=frame.buffer_data(), dtype=np.uint8, shape=(Height, Width))
+
+
+    cv2.imshow("Frame_8bits", frame_8bits)
+    k = cv2.waitKey(100)
+
+    if (frame_pixel_format == "BayerRG8" or frame_pixel_format == "BayerRG12" or frame_pixel_format == "BayerRG12Packed"):
+        colorImg = cv2.cvtColor(frame_8bits, cv2.COLOR_BAYER_RG2RGB )
+        cv2.imshow("Color_Image", colorImg)
+    elif (frame_pixel_format == "BayerGR8" or frame_pixel_format == "BayerGR12" or frame_pixel_format == "BayerGR12Packed"):
+        colorImg = cv2.cvtColor(frame_8bits, cv2.COLOR_BAYER_GR2RGB )
+        cv2.imshow("Color_Image", colorImg)
+    elif (frame_pixel_format == "RGB8Packed" or frame_pixel_format == "BGR8Packed"):
+        RGBImg = frame_8bits.reshape(Height, Width, 3)
+        colorImg = cv2.cvtColor(RGBImg, cv2.COLOR_BGR2RGB)
+        cv2.imshow("Color_Image", colorImg)
+
+    k = cv2.waitKey(0)
+
+
+camera.disarm()
+camera.close()
+Vimba().shutdown()
+
+
+def main():
+    vimba = Vimba()
+    vimba.startup()
+    vmFactory = vimba.camera_ids()
+    # Get connected cameras
+    cams = [vimba.camera(id) for id in vmFactory]
+    if len(cams) == 0:
+        raise OSError("No camera present.")
+    for idx, device in enumerate(vmFactory):
+        print("Device {} ID: {}".format(idx, device))
+    camera = vimba.camera(0)
+    camera.open()
+
+    pixel_format = camera.feature("PixelFormat")
+    # pixel_format.value = "BayerRG8"
+    # Note: Tested with at "Mono8" "BayerRG8" "BayerRG12" "BayerRG12Packed" "RGB8Packed" pixel format
+
+    camera.arm('SingleFrame')
+    image_Height = camera.feature("Height")
+    image_Width = camera.feature("Width")
+    Height = image_Height.value
+    Width = image_Width.value
+
+    # capture a single frame, more than once if desired
+    for idx in range(1):
+        frame = camera.acquire_frame()
+        print('frame No.{}'.format(frame.data.frameID))
+        camera_frame_size = len(frame.buffer_data())
+        frame_pixel_format = frame.pixel_format
+        print("Frame size: %d,  Image Resolution: %dx%d,  pixel_format: %s " % (camera_frame_size, Width, Height, frame_pixel_format))
+        data_bytes = frame.buffer_data()
+
+        if (frame_pixel_format == "Mono8" or frame_pixel_format == "BayerRG8" or frame_pixel_format == "BayerGR8"):
+            frame_8bits = np.ndarray(buffer=data_bytes, dtype=np.uint8, shape=(Height, Width))
+
+        elif (frame_pixel_format == "BayerRG12" or frame_pixel_format == "Mono10" or frame_pixel_format == "Mono12" or frame_pixel_format == "Mono14"):
+            data_bytes = np.frombuffer(data_bytes, dtype=np.uint8)
+            pixel_even = data_bytes[0::2]
+            pixel_odd = data_bytes[1::2]
+
+            # Convert bayer16 to bayer8 / Convert Mono12/Mono14 to Mono8
+            if (frame_pixel_format == "Mono14"):
+                pixel_even = np.right_shift(pixel_even, 6)
+                pixel_odd = np.left_shift(pixel_odd, 2)
+            elif (frame_pixel_format == "Mono10"):
+                pixel_even = np.right_shift(pixel_even, 2)
+                pixel_odd = np.left_shift(pixel_odd, 6)
+            else:
+                pixel_even = np.right_shift(pixel_even, 4)
+                pixel_odd = np.left_shift(pixel_odd, 4)
+            frame_8bits = np.bitwise_or(pixel_even, pixel_odd).reshape(Height, Width)
+
+        elif (frame_pixel_format == "BayerRG12Packed" or frame_pixel_format == "Mono12Packed" or frame_pixel_format == "BayerGR12Packed"):
+            data_bytes = np.frombuffer(data_bytes, dtype=np.uint8)
+            size = len(data_bytes)
+            index = []
+            for j in range(0, size, 3):
+                index.append(j+1)
+
+            data_bytes = np.delete(data_bytes, index)
+            frame_8bits = data_bytes.reshape(Height, Width)
+
+        elif (frame_pixel_format == "RGB8Packed" or frame_pixel_format == "BGR8Packed"):
+            frame_8bits = np.ndarray(buffer=frame.buffer_data(), dtype=np.uint8, shape=(Height, Width*3))
+
+        else:
+            # Note: wait to do -- other format, such as YUV411Packed, YUV422Packed, YUV444Packed
+            frame_8bits = np.ndarray(buffer=frame.buffer_data(), dtype=np.uint8, shape=(Height, Width))
+
+
+        cv2.imshow("Frame_8bits", frame_8bits)
+        k = cv2.waitKey(100)
+
+        if (frame_pixel_format == "BayerRG8" or frame_pixel_format == "BayerRG12" or frame_pixel_format == "BayerRG12Packed"):
+            colorImg = cv2.cvtColor(frame_8bits, cv2.COLOR_BAYER_RG2RGB )
+            cv2.imshow("Color_Image", colorImg)
+        elif (frame_pixel_format == "BayerGR8" or frame_pixel_format == "BayerGR12" or frame_pixel_format == "BayerGR12Packed"):
+            colorImg = cv2.cvtColor(frame_8bits, cv2.COLOR_BAYER_GR2RGB )
+            cv2.imshow("Color_Image", colorImg)
+        elif (frame_pixel_format == "RGB8Packed" or frame_pixel_format == "BGR8Packed"):
+            RGBImg = frame_8bits.reshape(Height, Width, 3)
+            colorImg = cv2.cvtColor(RGBImg, cv2.COLOR_BGR2RGB)
+            cv2.imshow("Color_Image", colorImg)
+
+        k = cv2.waitKey(0)
+        print("exit image %d display..." %idx)
+
+    camera.disarm()
+    camera.close()
+    cv2.destroyAllWindows()
+    time.sleep(0.5)
+    vimba.shutdown()
+    print("Exit Program...")
+
+if __name__ == '__main__':
+    main()
+