VOC: 基于 YOLO V8 的 PASCAL VOC 2007 目标检测项目项目简介VOC-VOLO-V8是一个基于 YOLO V8 对VOC 2007的目标检测项目专注于 PASCAL VOC 2007 数据集的处理、训练和预测。该项目通过将 PASCAL VOC 2007 数据集转换为 YOLO 格式并利用 YOLO V8 模型进行训练实现了高效的目标检测任务。数据集PASCAL VOC 2007 数据集PASCAL VOC 2007 是计算机视觉领域中一个著名的标准数据集包含 9963 张图片分为训练集5011 张和测试集4952 张涵盖 20 个类别如飞机、自行车、鸟、船、瓶子等。其标注信息以 XML 格式存储包含目标的边界框和类别标签。数据集处理项目中提供了以下功能下载数据集从官方 URL 下载 PASCAL VOC 2007 数据集。格式转换将原始的 VOC 格式XML转换为 YOLO 格式TXT包括边界框和类别标签的标准化。数据集划分将数据集划分为训练集和测试集并生成voc.yaml配置文件。数据集下载# 下载并解压数据集 download_t(VOC_URL, DATA_DIR) download_t(VOC_TEST_URL, DATA_DIR)环境依赖Python 环境Python 3.8 或更高版本PyTorch 1.10 或更高版本推荐使用 CUDA 支持Ultralytics YOLO 8.3.108 或更高版本安装依赖pip install ultralytics项目结构VOC-yolo/ ├── baseModel/ # 预训练模型 ├── dataset/ # 数据集存储目录 │ ├── output/ # 转换后的 YOLO 格式数据集 │ └── voc.yaml # 数据集配置文件 ├── predict/ # 预测结果保存目录 ├── runs/ # 训练结果保存目录 ├── .gitignore # Git 忽略文件 ├── datasetDownLoad.py # 数据集下载和转换脚本 ├── model_train_predict_ultralytics.ipynb # 训练和预测代码 ├── README.MD # 项目说明 └── voc.yaml # 数据集配置文件使用方法1. 数据集下载与转换运行以下代码下载 PASCAL VOC 2007 数据集并转换为 YOLO 格式# 转换 VOC 格式到 YOLO 格式 voc_to_yolo)2. 模型训练加载预训练模型并开始训练# 加载预训练模型 model YOLO(baseModel/yolov8n.pt) # 训练模型 model.train( datavoc.yaml, epochs30, batch64, imgsz800, devicecuda, optimizerAdamW, lr00.0001, augmentTrue, saveTrue, exist_okTrue )3. 模型预测运行以下代码进行预测# 预测图片 run_predict(dataset/output/images/test/000001.jpg, saveTrue, save_pathpredict/) # 预测视频 run_predict(dataset/output/video.mp4, saveTrue, save_pathpredict/) # 预测文件夹 run_predict(dataset/output/images/test/, saveTrue, save_pathpredict/) # 摄像头实时预测 run_predict(0)性能评估模型性能训练完成后模型在 PASCAL VOC 2007 测试集上的性能如下指标值mAP500.925mAP50-950.734每张图片推理时间7.5ms各类别性能类别PrecisionRecallmAP50mAP50-95aeroplane0.9720.9210.9830.812bicycle0.9200.8710.9480.741bird0.8690.8800.9370.719boat0.8700.7790.8760.648bottle0.8360.7480.8350.579bus0.9190.9080.9550.830car0.9040.8560.9430.745cat0.9390.9280.9690.851chair0.8680.7010.8360.616cow0.9110.8660.9510.740diningtable0.8980.7810.8940.711dog0.9370.9220.9720.832horse0.9250.9210.9620.809motorbike0.8810.8590.9270.732person0.9220.8460.9450.725pottedplant0.8500.7100.8350.561sheep0.9030.8660.9250.668sofa0.8850.7450.8850.746train0.9730.9360.9730.837tvmonitor0.9240.8770.9480.771参考资源PASCAL VOC 2007 官方网站Ultralytics YOLO 官方文档关于数据集下载链接问题由于网络原因PASCAL VOC 2007 数据集的下载链接可能无法正常解析。如果遇到问题请检查以下内容数据集VOCtrainval_06-Nov-2007.tar 和 VOCtest_06-Nov-2007.tar。尝试使用其他网络环境或代理工具。如果问题仍然存在可以考虑从其他镜像站点下载数据集。