这篇博文介绍一下YOLOv5自带的一个生成标签的方式
    
    
    
    
，前提是有一个已经训练好的权重！！    1    
    
    
    
  、权重可以来自官方提供的     


     


     


     


  ，比如v5的权重包含80类别 




 




 




 




 
，如果包含了你要训练的类别

   

   

   

   

，就可以用官方权重去打标签；2 


     


     


     


     
、或者你的数据集打好一部分标签     

     

     

     

 ，自己先训练出一个权重  




  




  




  




  ，用训练出的权重去检测剩下的图片


  


  


  


  


，生成标签文件等等    
    
    
    
  。

目录

1

 




 




 




 




、利用现有的权重去检测图片     
     
     
     
，得到标签文件

1.1修改相应参数

 2   

   

   

   

 、对生成的标签文件使用IableImg进行微调

 3  

     

     

     

     、将所有标签微调好了后   




   




   




   




   ，下面是对数据进行划分 


     


     


     


     
。

1


  




  




  




  




、利用现有的权重去检测图片



 



 



 



 


，得到标签文件

    找到YOLOv5的detect.py文件                    ，以我下面检测行人为例    




    




    




    




   ，可以去官网下载YOLOv5X.pt






















，因为官方权重包括行人类                    ，所有可以直接使用     



     



     



     



   ，这里建议下载最大的





















，因为检测精度高

 




 




 




 




。

1.1修改相应参数

  1  


  


  


  


、参数save-txt的末尾添加default = True
    
    
    
    
，这个参数就是保存检测的标签文件   

   

   

   

 。

parser.add_argument(--save-txt, action=store_true, help=save results to *.txt,default=True)

  2   
     
     
     
     、参数nosave的的末尾添加default = True     


     


     


     


  ，这个参数是是否保存检测后图片 




 




 




 




 
，设置不保存  

     

     

     

     。

parser.add_argument(--nosave, action=store_true, help=do not save images/videos,default=True)

  3



   




   




   




   



、然后官网权重是80类别

   

   

   

   

，我现在只想检测行人     

     

     

     

 ，可以在pycharm的Termina运行该detect.py

时加上--classes 0,就是只检测行人这个类


  




  




  




  




。

python detect.py --classes 0

   4 



 



 



 



、接下来去找刚才的运行结果  




  




  




  




  ，一般默认保存在run/detect/文件夹下


  


  


  


  


，存在一个labels文件     
     
     
     
，就是我们需要的标签文件  


  


  


  


。

 2                       、对生成的标签文件使用IableImg进行微调

  1




    




    




    




    


、因为用初始权重得到的标签的坐标框信息可能存在一些误差   




   




   




   




   ，所以还需要进行手动微调



 



 



 



 


，哈哈                    ，是不是又回到了最开始的打标签的时候    




    




    




    




   ，这样其实已经能节约很多时间了   
     
     
     
     。

  2



















、将最开始测试图片和得到的labels文件夹放在一起






















，使用labelImg打开该文件夹



   




   




   




   



。

打开labelImg之前                    ，先在labels文件夹下手动创建一个classes.txt文件     



     



     



     



   ，里面写上你的类别名称





















，防止labelImg的闪退 



 



 



 



。

  3                       、设置打开的图片文件夹
    
    
    
    
，设置保存标签的文件夹     


     


     


     


  ，如下图

   完成图片中1,2步后 




 




 




 




 
，第3步双击右侧图片

   

   

   

   

，即可对框进行修改                       。

  4



     



     



     



     

、打开自动保存功能     

     

     

     

 ，如下图

 3




















、将所有标签微调好了后  




  




  




  




  ，下面是对数据进行划分




    




    




    




    


。

   1    
    
    
    
  、这里提供一个将数据划分为如下格式的脚本


  


  


  


  


，也能用于v5的训练



















。图片如下

   2 


     


     


     


     
、脚本如下：只需要改main函数中     
     
     
     
，文件路径

 




 




 




 




、rate比列（训练集   




   




   




   




   ，验证集比列）                       。

import os import random import re import shutil def creat_files(): try: shutil.rmtree(./data//images) shutil.rmtree(./data/labels) except: pass try: os.makedirs(./data/images/train) os.makedirs(./data/images/val) os.makedirs(./data/labels/train) os.makedirs(./data/labels/val) except: pass def read_files(path): dir_list = os.listdir(path) for i in dir_list: #利用正则表达



 



 



 



 


，切割出图片ID image_id = re.match(r(.*)?.(jpg|png),i).group(1)#图片后缀为jpg                    ，png,可自行添加后缀格式 labels_name = image_id+.txt #由图片ID找到对应标签名 prob = random.randint(1, 100)#随机数prob if (prob < rate): # train dataset old_images_path = path_images + / + i new_images_path = ./data/images/train + / + i old_labels_path = path_labels + /+ labels_name new_labels_path = ./data/labels/train + / + labels_name shutil.copy(old_images_path, new_images_path) shutil.copy(old_labels_path, new_labels_path) else: # val dataset old_images_path = path_images + / + i new_images_path = ./data/images/val + / + i old_labels_path = path_labels + / + labels_name new_labels_path = ./data/labels/val + / + labels_name shutil.copy(old_images_path, new_images_path) shutil.copy(old_labels_path, new_labels_path) if __name__ == __main__: path_images = ./images #图片的路径 path_labels = ./labels #标签路径 rate = 90 # （1-100）之间    




    




    




    




   ，90即练集与验证集比例9:1






















，可以改成自己喜欢的比例 creat_files() read_files(path_images)

   3   

   

   

   

 、运行脚本会在代码根目录生成一个data文件                    ，里面划分好数据     



     



     



     



   ，如下：

    4  

     

     

     

     、可以直接把这个data文件夹剪切到YOLOv5中进行训练了



     



     



     



     

。