35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
import cv2
|
||
import numpy as np
|
||
|
||
cap = cv2.VideoCapture('/dev/video0', cv2.CAP_V4L2)
|
||
cap.set(cv2.CAP_PROP_FOURCC, cv2.VideoWriter_fourcc('B', 'G', 'R', '3'))
|
||
cap.set(cv2.CAP_PROP_FRAME_WIDTH, 1280)
|
||
cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 720)
|
||
|
||
ret, frame = cap.read()
|
||
if ret:
|
||
print(f"Captured frame shape: {frame.shape}")
|
||
|
||
# 解决形状问题，reshape frames to (720, 1280, 3)
|
||
frame = frame.reshape((720, 1280, 3)) # ensure the correct shape if frame is 1D
|
||
|
||
# 确保数据是有效的8位无符号整型
|
||
if frame.dtype != np.uint8:
|
||
frame = frame.astype(np.uint8)
|
||
|
||
# 输出调试信息
|
||
print(f"Frame dtype: {frame.dtype}, Frame shape after reshape: {frame.shape}")
|
||
|
||
# 尝试保存为JPEG
|
||
cv2.imwrite("hdmi_frame.jpg", frame)
|
||
print("Frame saved as JPEG")
|
||
|
||
# 可选：保存RAW数据
|
||
with open("hdmi_frame.raw", "wb") as f:
|
||
f.write(frame.tobytes())
|
||
print("RAW data saved")
|
||
else:
|
||
print("Capture failed")
|
||
|
||
cap.release()
|