使用python,将文件夹中使用labelme标注好的jpg图片和json文件中的shape水平翻转,且将生成的json文件的imagepath和imagedata与新生成的图片一一对应,并以确定的起始编号存放在新文件夹里
时间: 2024-01-22 18:21:16 浏览: 93
可以使用以下代码实现该功能:
```python
import os
import json
from PIL import Image
# 输入文件夹和输出文件夹
input_folder = 'path/to/input/folder'
output_folder = 'path/to/output/folder'
# 起始编号
start_index = 1
# 遍历所有图片和JSON文件
for filename in os.listdir(input_folder):
if filename.endswith('.jpg'):
# 读取图片
image_path = os.path.join(input_folder, filename)
image = Image.open(image_path)
image = image.transpose(Image.FLIP_LEFT_RIGHT)
# 生成新文件名和路径
new_filename = f'{start_index:06d}.jpg'
new_image_path = os.path.join(output_folder, new_filename)
# 保存图片
image.save(new_image_path)
# 读取对应的JSON文件
json_path = os.path.join(input_folder, f'{os.path.splitext(filename)[0]}.json')
with open(json_path, 'r') as f:
json_data = json.load(f)
# 更新JSON数据
json_data['imagePath'] = new_filename
json_data['imageData'] = None
for shape in json_data['shapes']:
for point in shape['points']:
point[0] = image.width - point[0]
# 生成新JSON文件名和路径
new_json_path = os.path.join(output_folder, f'{start_index:06d}.json')
# 保存新JSON文件
with open(new_json_path, 'w') as f:
json.dump(json_data, f)
# 更新起始编号
start_index += 1
```
上述代码使用PIL库来水平翻转图片,并使用json库来读取和修改JSON文件。在生成新的JSON文件时,将`imagePath`和`imageData`更新为新的图片文件名,并将所有的形状点进行水平翻转。最后,使用`json.dump`函数将更新后的JSON数据保存到新的JSON文件中。
阅读全文