我要对一个文件夹里的图片都进行一个同一个处理,将结果保存到csv中,同时我也要把图片的名称也对应保存到csv中去查看每张图片都处理结果,怎么写代码
时间: 2024-03-10 13:50:22 浏览: 11
你可以使用Python中的Pillow库来处理图片,使用csv库来保存结果。以下是一个简单的代码示例:
```python
from PIL import Image
import csv
import os
# 定义图片处理函数
def process_image(image_path):
img = Image.open(image_path)
# 在这里添加你的图片处理逻辑
result = img.size[0] * img.size[1] # 这里仅作示例,计算图片像素数
img.close()
return result
# 遍历文件夹中的所有图片并处理
folder_path = '/path/to/folder'
results = []
for filename in os.listdir(folder_path):
if not filename.endswith('.jpg'): # 如果不是jpg图片则跳过
continue
image_path = os.path.join(folder_path, filename)
result = process_image(image_path)
results.append((filename, result))
# 将结果保存到csv文件中
with open('image_results.csv', 'w', newline='') as csvfile:
writer = csv.writer(csvfile)
writer.writerow(['filename', 'result'])
for item in results:
writer.writerow(item)
```
以上代码会将指定文件夹中的所有jpg图片进行处理,并将每张图片的名称和处理结果保存到csv文件中。你可以根据需要修改图片处理逻辑和保存结果的方式。