世代网络网站建设设计,网站做生鲜线下推广建议,wordpress无法下载,网站横幅怎么制作教程小编在使用yolov5进行深度学习训练的时候#xff0c;训练了好几个epochs#xff0c;突然报了一个empty的错误导致训练程序终止#xff0c;还好小编只是在训练刚开始遇到这种问题#xff0c;如果训练了好多轮以后遇到此类问题#xff0c;那前面训练花费的时间全部都浪费了训练了好几个epochs突然报了一个empty的错误导致训练程序终止还好小编只是在训练刚开始遇到这种问题如果训练了好多轮以后遇到此类问题那前面训练花费的时间全部都浪费了那就非常难受了那我们可以从源头上改变这个问题。那我们如何改变呢我们需要根据空标签txt来删除对应的图片因为数据集有严格的一一对应原则代码如下
import osdef 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 或 .pngimage_file label_file.replace(.txt, .jpg) # 假设图片扩展名为 .jpg如果是 .png 则修改为 .pngimage_path os.path.join(image_dir, image_file)# 删除空标签文件print(fDeleting empty label file: {label_path})os.remove(label_path)# 检查对应的图片文件是否存在并删除if os.path.isfile(image_path):print(fDeleting corresponding image file: {image_path})os.remove(image_path)else:print(fCorresponding 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)把路径改一下以后训练可以继续了
