Python将指定label的标注框绘制到图片上

前言

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

前提条件

• 熟悉Python

相关介绍

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

实验环境

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

Python将指定label的标注框绘制到图片上

import os
import cv2
import json
import numpy as np


def draw_rect(img_path, coords, output_path):
    '''
    img = cv2.rectangle( img, pt1, pt2, color[, thickness[, lineType]] )
    参数说明：
        img：在其上面绘制图形的载体图像（绘图的容器载体，也称为画布、画板）。
        pt1 为矩形顶点。
        pt2 为矩形中与 pt1 对角的顶点。
        color：绘制形状的颜色。通常使用 BGR 模型表示颜色，
            例如，(0, 255, 0)表示绿色。对于灰度图像，只能传入灰度值。
            需要注意，颜色通道的顺序是 BGR，而不是 RGB。
        thickness：线条的粗细。
            默认值是 1，如果设置为-1，表示填充图形（即绘制的图形是实心的）。
        lineType：线条的类型，默认是 8 连接类型。
    '''
    img = cv2.imread(img_path)
    for coord in coords:
        cv2.rectangle(img, (coord[0], coord[1]),(coord[2], coord[3]), (0, 0, 255), 2)

    cv2.imwrite(output_path, img)


def json_rect(in_json_path,label_name):
    with open(in_json_path, "r", encoding='utf-8') as f:
        # json.load数据到变量json_data
        json_data = json.load(f)

    coords = []
    for i in json_data['shapes']:
        if i['label']==label_name:
            # print(i['points'])
            x1 = int(i['points'][0][0])
            y1 = int(i['points'][0][1])
            x2 = int(i['points'][1][0])
            y2 = int(i['points'][1][1])
            coords.append([x1,y1,x2,y2])

    return coords


if __name__ == "__main__":
    images_dir = './images/'
    jsons_dir = './jsons/'
    output_dir = 'output/'

    if not os.path.exists(output_dir):
        os.mkdir(output_dir)

    images_dir_list = os.listdir(images_dir)
    images_list = [i for i in images_dir_list if i.endswith(
        '.jpg') or i.endswith('.png') or i.endswith('.bmp')]
    # print(images_list)

    jsons_dir_list = os.listdir(jsons_dir)
    jsons_list = [i for i in jsons_dir_list if i.endswith('.json')]
    # print(jsons_list)

    label_name = input("请输入要绘制的label_name:")
    for image in images_list:
        img_path = images_dir + image
        in_json_path = jsons_dir + image[:-4] + '.json'
        output_path = output_dir + image
        # print(img_path,in_json_path)
        coords = json_rect(in_json_path,label_name)
        if coords == []:
            continue
        draw_rect(img_path, coords, output_path)

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