opencv读取图像集代码
时间: 2024-09-30 17:00:50 浏览: 38
在OpenCV中,读取图像集通常涉及到迭代一个文件夹并逐个打开其中的图片。这里是一个基本的Python示例,它使用os模块来遍历文件夹,cv2库来读取图像:
```python
import cv2
import os
def read_image_folder(path):
images = []
for filename in os.listdir(path): # 遍历文件夹中的所有文件
if filename.endswith('.jpg') or filename.endswith('.png'): # 只处理.jpg和.png格式的文件
img_path = os.path.join(path, filename)
image = cv2.imread(img_path) # 使用cv2.imread读取图像
if image is not None: # 检查是否成功读取
images.append(image)
else:
print(f"无法读取文件: {img_path}")
return images
# 调用函数并传入图像文件夹路径
image_dataset = read_image_folder('your_directory')
```
在这个例子中,`read_image_folder`函数会返回一个包含所有图片的列表。如果遇到无法读取的文件,会打印一条错误消息。
阅读全文