import json import os from glob import glob from labelme import utils from utils import * import cv2 def run(json_list: list) -> None: for json_path in json_list: with open(json_path, 'r', encoding="utf-8") as f: data = json.load(f) imageData = data["imageData"] imagePath = data["imagePath"] img = utils.img_b64_to_arr(imageData) if img.shape[2] == 4: img = cv2.cvtColor(img, cv2.COLOR_RGBA2BGR) else: img = img[:, :, ::-1] img_save_path = os.path.join(image_dir, imagePath) suffix = os.path.splitext(imagePath)[-1] write(img_save_path, img, suffix) def main() -> None: json_list = glob(f"{json_dir}\\*.*") run(json_list) if __name__ == '__main__': root_dir = r"文件名" json_dir = os.path.join(root_dir, "1") image_dir = os.path.join(root_dir, "2") os.makedirs(image_dir, exist_ok=True) main()
07-20
2539
10-16
1437