labelme批量转换json方法（5.2.1版本）

在anaconda的安装目录中找D:\anac\envs\(虚拟环境)\Lib\site-packages\labelme\cli

 将代码替换为

import argparse
import base64
import json
import os
import os.path as osp
import glob
import imgviz
import PIL.Image
from labelme.logger import logger
from labelme import utils


def process_single_json(json_file, out_dir=None):
data = json.load(open(json_file))
imageData = data.get("imageData")

if not imageData:
imagePath = os.path.join(os.path.dirname(json_file), data["imagePath"])
with open(imagePath, "rb") as f:
imageData = f.read()
imageData = base64.b64encode(imageData).decode("utf-8")
img = utils.img_b64_to_arr(imageData)

label_name_to_value = {"_background_": 0}
for shape in sorted(data["shapes"], key=lambda x: x["label"]):
label_name = shape["label"]
if label_name in label_name_to_value:
label_value = label_name_to_value[label_name]
else:
label_value = len(label_name_to_value)
label_name_to_value[label_name] = label_value
lbl, _ = utils.shapes_to_label(
img.shape, data["shapes"], label_name_to_value
)

label_names = [None] * (max(label_name_to_value.values()) + 1)
for name, value in label_name_to_value.items():
label_names[value] = name

lbl_viz = imgviz.label2rgb(
lbl, imgviz.asgray(img), label_names=label_names, loc="rb"
)

if out_dir is None:
out_dir = osp.basename(json_file).replace(".", "_")
out_dir = osp.join(osp.dirname(json_file), out_dir)
else:
out_dir = out_dir
if not osp.exists(out_dir):
os.mkdir(out_dir)

PIL.Image.fromarray(img).save(osp.join(out_dir, "img.png"))
utils.lblsave(osp.join(out_dir, "label.png"), lbl)
PIL.Image.fromarray(lbl_viz).save(osp.join(out_dir, "label_viz.png"))

with open(osp.join(out_dir, "label_names.txt"), "w") as f:
for lbl_name in label_names:
f.write(lbl_name + "\n")

logger.info("Saved to: {}".format(out_dir))


def batch_process_json_files(json_dir, output_dir=None):
# 获取json_dir目录下所有JSON文件的路径
json_files = glob.glob(os.path.join(json_dir, '*.json'))

for json_file in json_files:
# 调用process_single_json()函数进行处理
file_name = osp.splitext(osp.basename(json_file))[0]
file_out_dir = osp.join(output_dir, file_name)

process_single_json(json_file, file_out_dir)

if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("json_dir", help="包含JSON文件的输入目录")
parser.add_argument("-o", "--output_dir", default=None, help="输出目录")
args = parser.parse_args()

json_dir = args.json_dir
output_dir = args.output_dir

if output_dir is None:
output_dir = osp.join(json_dir, "output")
if not osp.exists(output_dir):
os.makedirs(output_dir)

batch_process_json_files(json_dir, output_dir)

在anaconda prompt中cd/d 切换到D:\anac\envs\(虚拟环境)\Lib\site-packages\labelme\cli中输入：

python json_to_dataset.py xxx(json文件所在路径，保证里面只有json文件) -o xxx(输出文件夹)

代码效果如下：

每张图都会生成一个文件夹，每个文件夹有4个文件，对应原图、标签图、标签名称以及可视化叠加图