Python修改图片格式以及相对应labelme标注的Json文件

前言

• 本文是个人使用PythonPython处理文件的电子笔记，由于水平有限，难免出现错漏，敬请批评改正。
• 更多精彩内容，可点击进入我的个人主页查看

前提条件

• 熟悉Python

相关介绍

• Python是一种跨平台的计算机程序设计语言。是一个高层次的结合了解释性、编译性、互动性和面向对象的脚本语言。最初被设计用于编写自动化脚本(shell)，随着版本的不断更新和语言新功能的添加，越多被用于独立的、大型项目的开发。
• Python OS模块是负责程序与操作系统的交互，提供了访问操作系统底层的接口和非常丰富的方法用来处理文件和目录。

实验环境

• Python 3.x （面向对象的高级语言）

Python修改图片格式以及相对应labelme标注的Json文件

{
  "version": "5.0.1",
  "flags": {
    
  },
  "shapes": [
    {
      "label": "1",
      "points": [
        [
          5404.0,
          1606.666666666667
        ],
        [
          5557.333333333334,
          2166.666666666667
        ]
      ],
      "group_id": null,
      "shape_type": "rectangle",
      "flags": {
        
      }
    }
  ],
  "imagePath": "1.bmp",
  "imageData": null,
  "imageHeight": 4000,
  "imageWidth": 8320
}

代码实现

以bmp图片转换成jpg图片为例并修改"imagePath"的内容为例，代码如下。

import os
import cv2
import json

def bmp2jpg(in_img_path,out_dir_name): # .bmp -> .jpg
    # img = cv2.imread(in_img_path) # 彩色图片，位深24
    img = cv2.imread(in_img_path,0) # 黑白图片，位深8
    out_img_path = out_dir_name + in_img_path.split('/')[1][:-4] +'.jpg'
    # print(out_img_path)
    cv2.imwrite(out_img_path,img)


def alter_json(in_json_path,out_dir_name):
    file_in = open(in_json_path, "r", encoding='utf-8')
    # json.load数据到变量json_data
    json_data = json.load(file_in)
    # 修改json中的imagePath数据内容
    json_data["imagePath"] = json_data['imagePath'][:-4] + '.jpg'
    file_in.close()
    out_json_path = out_dir_name + in_json_path.split('/')[1]
    # 创建一个写文件
    file_out = open(out_json_path, "w", encoding='utf-8')
    # 将修改后的数据写入文件
    file_out.write(json.dumps(json_data))
    file_out.close()

if __name__=="__main__":
    in_dir_name = 'origin/' # 图片和json所在文件夹
    out_dir_name = 'output/'
    all_name_list = os.listdir(in_dir_name) # 获取文件夹内所有文件名列表

    # 获取所有bmp图片，.bmp -> .jpg
    image_name_list = [i for i in all_name_list if i.endswith('.bmp')] # 图片路径列表
    print(image_name_list)
    for image_name in image_name_list: # 遍历图片, .bmp -> .jpg
        in_img_path = in_dir_name + image_name
        bmp2jpg(in_img_path,out_dir_name)
        # os.remove(in_img_path) # 删除原始bmp图片
    
    # 获取所有json文件，并修改
    json_name_list = [i for i in all_name_list if i.endswith('.json')] # 图片路径列表
    print(json_name_list)
    for json_name in json_name_list: # 遍历json文件并修改
        in_json_path = in_dir_name + json_name
        alter_json(in_json_path,out_dir_name)

输出结果

{
  "version": "5.0.1",
  "flags": {
    
  },
  "shapes": [
    {
      "label": "1",
      "points": [
        [
          5404.0,
          1606.666666666667
        ],
        [
          5557.333333333334,
          2166.666666666667
        ]
      ],
      "group_id": null,
      "shape_type": "rectangle",
      "flags": {
        
      }
    }
  ],
  "imagePath": "1.jpg",
  "imageData": null,
  "imageHeight": 4000,
  "imageWidth": 8320
}

更多精彩内容，可点击进入Python日常小操作专栏或我的个人主页查看