帮我把下面这个代码的pic_path改为遍历文件夹中的所有图片import csv from IPython import display pic_path = "./期末作业-面部数据/face_images/a1/img_518.jpg" main(pic_path) result = main(pic_path) # 调用主函数获取结果 # 将结果转换成字典列表 result_dict = json.loads(result)["result"]["face_list"] output_list = [] for item in result_dict: output_list.append({ "file_name": item["face_token"], "emotion": item["emotion"]["type"] }) # 将结果保存为csv文件 with open("output.csv", "a", newline="") as f: writer = csv.DictWriter(f, fieldnames=["face_token", "emotion"]) #writer.writeheader() writer.writerows(output_list)
时间: 2023-07-19 19:30:25 浏览: 97
ipython-autoimport:在IPython中自动导入缺少的模块
好的,以下是修改后的代码:
```
import csv
import os
import json
from IPython import display
pic_folder_path = "./期末作业-面部数据/face_images/a1"
output_list = []
for filename in os.listdir(pic_folder_path):
if filename.endswith(".jpg"):
pic_path = os.path.join(pic_folder_path, filename)
result = main(pic_path) # 调用主函数获取结果
result_dict = json.loads(result)["result"]["face_list"]
for item in result_dict:
output_list.append({
"face_token": item["face_token"],
"emotion": item["emotion"]["type"]
})
# 将结果保存为csv文件
with open("output.csv", "a", newline="") as f:
writer = csv.DictWriter(f, fieldnames=["face_token", "emotion"])
#writer.writeheader()
writer.writerows(output_list)
```
这里我们使用了 `os.listdir()` 函数来遍历指定文件夹中的所有文件,然后通过判断文件名是否以 `.jpg` 结尾来筛选出图片文件。之后就可以将图片文件路径传入 `main` 函数获取结果,并将结果保存在 `output_list` 列表中。最后,将整个列表写入到 `output.csv` 文件中。
阅读全文