yolo自动标注时缺失的txt文件批量创建之脚本

原创

已于 2023-03-23 15:50:33 修改 · 676 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#python #YOLO #目标检测

于 2023-03-23 15:28:56 首次发布

使用yolov5做推理保存yolo格式txt文件时，若图片无检测目标，yolov5不保存空txt文件，会致训练报错。为此编写Python脚本，在不影响已有txt文件的情况下，批量创建缺失的空txt文件，保证txt与图片数量一致。

一、脚本功能

使用yolov5做推理并保存yolo格式的txt文件时，当图片上没有检测到目标时，yolov5将不会保存空的txt文件，导致txt文件缺失，yolo训练时报错。

所以写了个简单的小脚本，在不影响原来已存在txt文件的前提下，来批量创建缺失的空txt文件，确保txt文件的总数量与图片的总数量一致。

二、脚本源码

# -*- coding: utf-8 -*-

import os

img_list = []
txt_list = []
def img_path(image_path):
    for i in os.listdir(image_path):
        path = i.split(".")[0]
        img_list.append(path)
        # print(path)

def txt_path(label_path):
    for i in os.listdir(label_path):
        path = i.split(".")[0]
        txt_list.append(path)

def create_txt(label_path,txt):
    txt_txt = list(txt)
    for path in txt_txt:
        path_data = os.path.join(label_path, path) + '.txt'
        print(path_data)
        with open(path_data, 'w') as f:
            f.write("")
            f.close()
    pass


if __name__ == '__main__':
    image_path =