qq_51703448的博客

原创 【yolov5数据集分类】根据label里面的txt标签，将image图片分类到对应的文件夹下面

这段Python代码实现了对YOLO数据集的图片分类整理。程序将存储在images文件夹中的图片，根据labels文件夹中对应的标签内容（0=cat，1=dog，2=bird），自动分类到相应的子文件夹中。主要功能包括：读取标签文件内容建立图片-类别映射字典、创建分类文件夹结构、将图片移动到对应类别的子文件夹。代码还包含错误处理机制，如跳过空标签文件、处理文件权限问题等。使用时需设置图片路径(image_path)、标签路径(label_path)和目标分类路径(move_to_path)，并定义class