软件系统网站建设南宁推广公司

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

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

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

import os
import numpy as np
import halcon as haimport json
import shutilfrom qtconsole.mainwindow import background###逻辑是这样的
# 初始化时 输入 labelme标注的文件夹目录，和 类别列表
# 在项目目录下生成两个文件夹，一个是test_images, 一个是对应的marks_folder也就是标签图片
class gen_marks_image():def __init__(self,path,label_list):   #初始化时，需传入文件的路径self.path = pathself.label_list = label_listdef 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_filesdef 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的columnrow_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()