玩要导入一个文件夹下数据集方法图片有文件big的
时间: 2024-10-17 15:05:05 浏览: 26
不同字符编码集的文件读取
在Python中,如果你想要导入一个文件夹下的所有图片数据集,你可以使用`os`模块来遍历文件夹并加载图片。`PIL`库可以用于读取图片。这里有一个基本的示例,假设你的图片都在名为"big"的文件夹下:
```python
import os
import cv2
import numpy as np
def load_images_from_folder(folder_path):
images = []
for filename in os.listdir(folder_path): # 遍历文件夹内容
if filename.endswith(('.jpg', '.jpeg', '.png', '.bmp')): # 检查是否是图片文件
img_path = os.path.join(folder_path, filename) # 获取完整路径
try:
image = cv2.imread(img_path) # 使用OpenCV读取图片
# 如果需要,转换为RGB格式(因为cv2默认为BGR)
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
# 进行预处理(如缩放、裁剪等)
processed_image = process_image(image)
images.append(processed_image) # 添加到列表中
except Exception as e:
print(f"Error loading {img_path}: {e}")
return images
# 自定义的图像预处理函数
def process_image(image):
# ... (对图像进行你需要的操作)
return image
folder_path = 'path_to_your_big_folder'
images_data = load_images_from_folder(folder_path)
# ... (后续操作,比如将图片数据存储为numpy数组或DataFrame)
阅读全文