64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
# camera_capture.py
|
||
import os
|
||
from pypylon import pylon
|
||
import cv2
|
||
from datetime import datetime
|
||
|
||
|
||
def search_get_device():
|
||
"""搜索并返回 Basler GigE 相机对象"""
|
||
tl_factory = pylon.TlFactory.GetInstance()
|
||
for dev_info in tl_factory.EnumerateDevices():
|
||
if dev_info.GetDeviceClass() == 'BaslerGigE':
|
||
print(f"Found Basler GigE Camera: {dev_info.GetModelName()} IP: {dev_info.GetIpAddress()}")
|
||
return pylon.InstantCamera(tl_factory.CreateDevice(dev_info))
|
||
raise EnvironmentError("没有找到 Basler GigE 相机")
|
||
|
||
|
||
def grab_image(save=True):
|
||
"""
|
||
采集一张图像并返回 numpy ndarray（BGR）
|
||
save=True 时自动保存 PNG 文件到 camera/ 文件夹
|
||
"""
|
||
# 创建保存目录
|
||
save_dir = "camera"
|
||
os.makedirs(save_dir, exist_ok=True)
|
||
|
||
cam = search_get_device()
|
||
cam.Open()
|
||
|
||
# 设置图像转换器（BGR8 → OpenCV格式）
|
||
converter = pylon.ImageFormatConverter()
|
||
converter.OutputPixelFormat = pylon.PixelType_BGR8packed
|
||
converter.OutputBitAlignment = pylon.OutputBitAlignment_MsbAligned
|
||
|
||
cam.StartGrabbing(pylon.GrabStrategy_LatestImageOnly)
|
||
|
||
grabResult = cam.RetrieveResult(5000, pylon.TimeoutHandling_ThrowException)
|
||
|
||
if grabResult.GrabSucceeded():
|
||
image = converter.Convert(grabResult)
|
||
img = image.GetArray() # OpenCV 可处理的 ndarray
|
||
print("成功采集图像:", img.shape)
|
||
|
||
# 保存 PNG
|
||
if save:
|
||
filename = os.path.join(save_dir, f"Image_{datetime.now().strftime('%Y%m%d_%H%M%S')}.png")
|
||
cv2.imwrite(filename, img)
|
||
print("已保存:", filename)
|
||
|
||
else:
|
||
print("采集失败:", grabResult.ErrorCode, grabResult.ErrorDescription)
|
||
img = None
|
||
|
||
grabResult.Release()
|
||
cam.Close()
|
||
return img
|
||
|
||
|
||
if __name__ == "__main__":
|
||
img = grab_image(save=True)
|
||
if img is not None:
|
||
print("图像类型:", type(img))
|
||
print("图像尺寸:", img.shape)
|