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()