目标检测脚本之yolopose 多分类标签生成

目标检测脚本之yolopose 多分类标签生成

一、需求分析

yolopose 在训练多分类模型时，需要将其它不包含关键点类别的数据标签文件整理成与人型

二、需求实现

1. 使用(0，0，0)填充例如 猫、狗、自行车等类别的17个关键点
2. 合并人型类别和之前填充的其他类别的数据

# # 一、填充非关键点标签
import os
import shutil

label_path = ""  # 未包含人型关键点的标签

label_save_path = ""  #存取填充后的标签

if not os.path.exists(label_save_path):
    os.makedirs(label_save_path)


label_files = os.listdir(label1_path)


for txt in label_files:
    txt_path = os.path.join(label1_path,txt1)

    save_txt_path = os.path.join(label_save_path,txt1)


    save_text = []

    with open(txt_path,'r') as file,open(save_txt_path,'w+') as save_file:

        #获取类别的标签记录，并用0补齐关键点