| # import cv2 | |
| # # 必须加上 cv2.IMREAD_UNCHANGED，原封不动地读取 | |
| # img = cv2.imread('/home/wenting/zr/wt_dataset/LIDC_IDRI/img_complex_fb_256/LIDC-IDRI-0001/0002.png', cv2.IMREAD_UNCHANGED) | |
| # # 通过查看数据的类型 (dtype) 来判断 | |
| # if img.dtype == 'uint8': | |
| # print("这是一张 8 位图像 (像素值范围 0-255)") | |
| # elif img.dtype == 'uint16': | |
| # print("这是一张 16 位图像 (像素值范围 0-65535)") | |
| # else: | |
| # print(f"其他格式: {img.dtype}") | |
| # print(f"图像形状: {img.shape}") | |
| # # 如果是 (H, W)，说明是单通道灰度图；如果是 (H, W, 3)，说明是彩色图 | |
| import cv2 | |
| def check_image_channels_cv2(image_path): | |
| # 使用 IMREAD_UNCHANGED 保持图像原有通道数读取 | |
| img = cv2.imread(image_path, cv2.IMREAD_UNCHANGED) | |
| if img is None: | |
| return "错误：无法读取图像，请检查路径。" | |
| # 判断维度长度 | |
| if len(img.shape) == 2: | |
| return "该图像是单通道（灰度图）。" | |
| elif len(img.shape) == 3: | |
| channels = img.shape[2] | |
| if channels == 3: | |
| return "该图像是三通道（RGB/BGR）。" | |
| elif channels == 4: | |
| return "该图像是四通道（包含透明通道的 RGBA/BGRA）。" | |
| else: | |
| return f"该图像有 {channels} 个通道。" | |
| else: | |
| return "未知的图像格式。" | |
| # 测试脚本 | |
| result = check_image_channels_cv2("/home/wenting/zr/dataset/OmniGenCXR_inference/outputs/1024_256_7events_lora6000_50steps/0a6a34da-654c2878-8cc9db4d-0787bc5d-3dccbe79.png") | |
| print(result) |