用python将images中的base、novel、val文件里的图片生成.json文件
时间: 2023-05-23 07:07:34 浏览: 88
A:
你好!以下是用Python将images中的base、novel、val文件里的图片生成.json文件的代码:
```python
import os
import json
# 定义存储图片信息的字典
images_dict = {'base': [], 'novel': [], 'val': []}
# 遍历文件夹,获取图片信息
for image_type in images_dict.keys():
image_folder = os.path.join('images', image_type)
for file_name in os.listdir(image_folder):
if file_name.endswith('.jpg') or file_name.endswith('.png'):
file_path = os.path.join(image_folder, file_name)
image_info = {'filename': file_name, 'filepath': file_path}
images_dict[image_type].append(image_info)
# 写入json文件
with open('images_info.json', 'w') as f:
json.dump(images_dict, f)
```
此代码会遍历images文件夹下的base、novel、val三个子文件夹,然后将每个子文件夹中的图片信息存储到字典images_dict中。最后,将整个字典写入名为images_info.json的文件中。
希望能帮到你!
阅读全文