目标检测之将bbox绘制到图片上

绘制bbox

文件夹内容

其中train1里面放的是待绘制的图片
train1_crop.json里面保存的是train1文件夹里数据的信息，三个指标：name,defect_name,bbox
例如：

image用来保存绘制好框图的图片
代码如下：

import os
import json
from PIL import Image
import cv2

train1_json_path = '/media/lxg/D/keyan/tianchi/guangdong1/data_augment/data_crop/train1_crop.json'
the_image_root_path = '/media/lxg/D/keyan/tianchi/guangdong1/data_augment/data_crop/train1_crop/'
the_new_image_draw_path = '/media/lxg/D/keyan/tianchi/guangdong1/data_augment/data_crop/draw_bbox/image/'
defect_name2label = {
    '破洞': 1, '水渍': 2, '油渍': 2, '污渍': 2, '三丝': 3, '结头': 4, '花板跳': 5,