关于yolov5遇到空标签导致训练暂停的解决
小编在使用yolov5进行深度学习训练的时候，训练了好几个epochs，突然报了一个empty的错误导致训练程序终止，还好小编只是在训练刚开始遇到这种问题，如果训练了好多轮以后遇到此类问题，那前面训练花费的时间全部都浪费了，那就非常难受了，那我们可以从源头上改变这个问题。那我们如何改变呢，我们需要根据空标签txt来删除对应的图片，因为数据集有严格的一一对应原则，代码如下
import os
def remove_empty_label_and_image(image_dir, label_dir):
# 遍历标签文件夹中的所有 .txt 文件
for label_file in os.listdir(label_dir):
label_path = os.path.join(label_dir, label_file)
# 检查标签文件是否为空
if os.path.isfile(label_path) and label_file.endswith('.txt') and os.path.getsize(label_path) == 0:
# 获取对应的图片文件名（去掉 .txt 扩展名并加上图片的扩展名，例如 .jpg 或 .png）
image_file = label_file.replace('.txt', '.jpg') # 假设图片扩展名为 .jpg，如果是 .png 则修改为 .png
image_path = os.path.join(image_dir, image_file)
# 删除空标签文件
print(f"Deleting empty label file: {label_path}")
os.remove(label_path)
# 检查对应的图片文件是否存在并删除
if os.path.isfile(image_path):
print(f"Deleting corresponding image file: {image_path}")
os.remove(image_path)
else:
print(f"Corresponding image file not found for label: {label_path}")
# 使用实际的图像和标签目录路径
image_directory = 'your path' # 替换为你的图片文件夹路径
label_directory = 'your path' # 替换为你的标签文件夹路径
remove_empty_label_and_image(image_directory, label_directory)
把路径改一下以后，训练可以继续了