python中OpenCV读取文件夹中的图片名、图片通道数量保存为表格文件,要求第一列为图片名、第二列为通道数量
时间: 2024-09-26 13:15:16 浏览: 118
在Python中,你可以使用OpenCV (cv2) 和 pandas 库来实现这个任务。首先,确保已经安装了这两个库,如果没有可以使用pip安装:
```bash
pip install opencv-python-headless pandas
```
下面是一个示例脚本,它会遍历指定文件夹中的所有图片,获取每张图片的名称以及通道数(对于BGR图像,通道数通常是3),然后将信息存储到CSV文件中:
```python
import cv2
import os
import pandas as pd
# 定义文件夹路径
folder_path = "your_folder_path"
# 创建一个空列表来存储数据
image_data = []
# 遍历文件夹中的所有文件
for filename in os.listdir(folder_path):
if filename.endswith(('.jpg', '.png', '.bmp')): # 只处理常见的图片格式
img = cv2.imread(os.path.join(folder_path, filename), cv2.IMREAD_UNCHANGED)
# 获取通道数量,对于BGR图像,通道数为3
channels = len(img.shape)
image_data.append([filename, channels])
# 将数据转换为pandas DataFrame并保存为CSV
data_df = pd.DataFrame(image_data, columns=['Image Name', 'Channels'])
data_df.to_csv('image_channels.csv', index=False)
```
在这个例子中,你需要替换`"your_folder_path"`为你想要读取图片的实际文件夹路径。
阅读全文