给我一个python代码用cv2的方法将一个文件夹里的所有含有坐标信息的json文件,转化为可用于分割训练的标签二值化黑白png图片且命名方式不变
时间: 2023-04-01 10:04:16 浏览: 81
以下是一个可能的 Python 代码示例,用于将一个文件夹中的所有包含坐标信息的 JSON 文件转换为可用于分割训练的标签二值化黑白 PNG 图像,同时保持命名方式不变:
```python
import os
import cv2
import json
# 定义输入和输出文件夹路径
input_folder = "/path/to/json/files"
output_folder = "/path/to/output/png/files"
# 遍历输入文件夹中的所有 JSON 文件
for filename in os.listdir(input_folder):
if filename.endswith(".json"):
# 读取 JSON 文件中的坐标信息
with open(os.path.join(input_folder, filename), "r") as f:
data = json.load(f)
coords = data["coords"] # 假设坐标信息存储在 "coords" 字段中
# 创建一个空白的黑白图像
img = np.zeros((height, width), dtype=np.uint8)
# 在图像上绘制坐标信息
for coord in coords:
x, y = coord["x"], coord["y"] # 假设坐标信息存储在 "x" 和 "y" 字段中
img[y, x] = 255 # 将对应像素点设为白色
# 将图像保存为 PNG 文件,命名方式不变
output_filename = os.path.join(output_folder, filename.replace(".json", ".png"))
cv2.imwrite(output_filename, img)
```
请注意,这只是一个示例代码,具体实现可能需要根据实际情况进行调整。
阅读全文