关于使用yolov5的一些小细节处理

准备工作参考大佬文章：

大佬文章点这

1，如何在识别图片之后获得标签，方便用这个标签来做训练集。

在detect.py中加入框选的内容，这样运行后会自动生成标签，可以节省大量获取标签的时间

 2.识别后的标签会在末尾多出识别度的数字，怎么删除呢，话不多说上源码

from os import listdir

txtPath = 'D:\重要文件\python项目\图像识别\yolov5-master\\runs\detect\\txt'
txts = listdir(txtPath)
print(len(txts))
for txtid in txts:
    txtFile = open(txtPath + '/' + txtid, 'r+')

     # print(txtFile.read())
    txt = txtFile.read().split(' ')[:5]
    txtFile.close()
    txtFile = open(txtPath + '/' + txtid, 'w+')
    for num in txt:
        txtFile.write(num + ' ')

3.问题又来了，我们知道yolov5训练集图片和标签一一对应的，没有识别出的图片就没有标签，怎么把多于的图片删除达成图片和标签一一对应的效果呢，话不多说上代码。

import os

import shutil


def delete_file(dirname,list):
    ToDirName='C:/Users/King/Desktop/羽绒服 - 顺序版1'
    filelist = os.listdir(dirname)
    for file in filelist:
        oldpath = os.path.join(dirname, file)
        filename = os.path.splitext(file)[0]  # 文件名
        filetype = os.path.splitext(file)[1]  # 文件类型
        for compareName in list:

            if filename == compareName.rstrip('.txt'):
                middleName=compareName.rstrip('.txt')+'.jpg'
                newpath=os.path.join(dirname, middleName)
                topath=os.path.join(ToDirName, middleName)
                shutil.move(newpath,topath)
                print(compareName,'就是这个')


def getNameList(dirname):
    list=[]
    filelist = os.listdir(dirname)
    for file in filelist:
        oldpath = os.path.join(dirname, file)
        list.append(file)

    print(list)
    return list

list=getNameList("D:/重要文件/python项目/图像识别/yolov5-master/TaoBao/txt")
delete_file('C:/Users/King/Desktop/羽绒服 - 顺序版',list)