這一段代碼封裝了一個類，需要傳統一個圖片和標注后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()