Yolov5偷懒用算法生成负样本label文件

最新推荐文章于 2024-06-24 15:03:51 发布

WsToITL

最新推荐文章于 2024-06-24 15:03:51 发布

阅读量1.1k

点赞数 4

文章标签： python 目标检测

本文链接：https://blog.csdn.net/qq_28593177/article/details/129801884

版权

文章讲述了在处理大规模数据集时，面对大量负样本标签的困境。作者提出了一种简化方法，使用Python脚本批量创建表示负样本的空txt文件，以减少手动标注的工作量。脚本遍历指定的负样本图片路径，为每个图片生成对应的空标签文件。

摘要由CSDN通过智能技术生成

在学习yolov5中遇到了数据集标签的问题，巨大的数据集标定，谁看了都害怕，一般来说少量的负样本自己创建就行，但当数据集数量大时，负样本的数量也会稍微增加，多一点点就不想干了，想了想不如用简单的算法偷懒，毕竟负样本只是empty，创建空txt文件即可

#coding=utf-8 
import os
def text_create(name):
    desktop_path = "F:\\data\\Test\\"  # 创建的label文件的存放路径
    full_path = desktop_path + name + '.txt'
    file = open(full_path, 'w')
    file.close()


IamgesPath = 'C:\\Users\\Administrator\\Desktop\\falseIamge'          #负样本的地址
for i,j,k in os.walk(IamgesPath):
    for s in k:                                                       #其它图片格式稍微修改一下即可
        s=s.strip('.jpg')
        text_create(s)
        print(s)