【亲测免费】 小目标烟头检测VOC格式数据集和data.yaml配置文件

内容概要

本仓库提供了一个用于小目标烟头检测的VOC格式数据集，以及相关的配置文件。数据集包含了2385张图片，每张图片中都含有烟头。此外，仓库中还提供了split_data.py、xml2txt.py和data.yaml三个文件，方便用户进行数据处理和模型训练。

文件说明

1. 图片文件：

   • 包含2385张图片，每张图片中都标注了烟头的位置。

2. split_data.py：

   • 用于将数据集划分为训练集和验证集的脚本。用户可以根据需要调整划分比例。

3. xml2txt.py：

   • 用于将VOC格式的标注文件（.xml）转换为YOLO格式的标注文件（.txt）的脚本。

4. data.yaml：

   • 配置文件，包含了数据集的路径、类别信息等，方便用户在训练模型时使用。

使用说明

1. 数据集划分：

   • 运行split_data.py脚本，将数据集划分为训练集和验证集。

2. 标注格式转换：

   • 运行xml2txt.py脚本，将VOC格式的标注文件转换为YOLO格式的标注文件。

3. 模型训练：

   • 使用data.yaml配置文件进行模型训练，配置文件中包含了数据集的路径和类别信息。

注意事项

• 请确保在运行脚本前，已安装所需的Python库。
• 数据集中的图片和标注文件已按VOC格式组织，用户可以根据需要进行调整。

贡献

欢迎大家提出改进建议或贡献代码，共同完善这个数据集和相关工具。