python+halcon 解读labelme标注生成marksimage

这一段代码封装了一个类，需要传统一个图片和标注后json文件所在的地址，标注的选项是polygon，主要是用于unet深度学习网络

在初始化时需要输入文件（image+jeson）路径，多分类任务的label_list。会在项目目录下生成一个test_images文件夹，和marks_folder的文件夹

marks_image的制作使用halcon完成的，方便后期自己项目使用。欢迎交流

import os
import numpy as np
import halcon as ha

import json
import shutil

from qtconsole.mainwindow import background


###逻辑是这样的
# 初始化时 输入 labelme标注的文件夹目录，和 类别列表
# 在项目目录下生成两个文件夹，一个是test_images, 一个是对应的marks_folder也就是标签图片
class gen_marks_image():
    def __init__(self,path,label_list):   #初始化时，需传入文件的路径
        self.path = path
        self.label_list = label_list

    def get_list_in_path(self):
        name_files = os.listdir(self.path)
        json_files = []
        for name in name_files:
            if name.endswith('.json'):
                json_files.append(os.path.join(self.path, name))

        return json_files

    def create_folder(self,json_files):
        image_folders = "test_images"
        marks_folder = "marks_folder"
        if not os.path.exists(image_folders):
            os.mkdir(marks_folder)
        if not os.path.exists(marks_folder):
            os.mkdir(marks_folder)

        for json_1 in json_files:
            json_name=os.path.split(json_1)[-1]
            image_path = os.path.join(self.path, json_name.replace('.json','.jpg'))
            shutil.copy(image_path, image_folders)   #完成了对itest_image向test_images文件夹的复制

            ###根据原图生成以0填充的灰度图
            image00=ha.read_image(image_path)
            width,height =ha.get_image_size(image00)
            image_marks0 = ha.gen_image_const("byte",int(width[0]),int(height[0]))

            ###开始绘制marks_image图
            with open(json_1,"r") as f:
                json_data = json.load(f)
            print(json_data["shapes"])
            for shape in json_data["shapes"]:
                label0 =shape["label"]
                ####在这里我们输入一个label的list列表，如果是多分类的话，通过label在label_list中的位置定义它的灰度值
                gray =self.label_list.index(label0)  #car对应1，dog对应2 作为他们的灰度值

                ####在这里开始绘制区域
                points = shape["points"]
                row_points=[]
                col_points=[]
                for point in points:
                    #这里注意lableme的x——对halcon的column
                    row_points.append(int(point[1]))
                    col_points.append(int(point[0]))
                    print(row_points,col_points)

                #在这里要注意，polygon的首尾要接上。
                row_points.append(row_points[0])
                col_points.append(col_points[0])
                region = ha.gen_region_polygon(row_points,col_points)

                print(gray)
                image_marks0 = ha.paint_region(region,image_marks0,gray,'fill')
                ha.write_image(image_marks0,'jpg',0,marks_folder+"/"+json_name.replace('.json','.jpg'))



    def forward(self):
        json_files = self.get_list_in_path()
        print(json_files)
        self.create_folder(json_files)


#使用过程其实很简单
if __name__ == '__main__':
    path = r'C:\Users\Administrator\Desktop\test_image'
    label_list = ['back', 'car', 'dog', 'cat']
    gen_marks_image(path,label_list).forward()