【YOLO】使用YOLOv3训练及测试自己的数据集

最新推荐文章于 2025-05-28 00:00:00 发布

EastPlain

最新推荐文章于 2025-05-28 00:00:00 发布

阅读量1.9k

点赞数

CC 4.0 BY-SA版权

分类专栏： YOLO # YOLOv3 # labelImg

本文链接：https://blog.csdn.net/East_Plain/article/details/108151712

本文详细介绍了如何在Ubuntu 16.04环境下，使用Python 3.7.0和PyTorch 1.6.0，基于YOLOv3训练和测试自己的数据集。从修改YOLOv3项目、准备数据集、修改配置文件、下载预训练权重到进行训练和测试，每个步骤都有清晰的操作说明，并提供了参考资料链接。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使用YOLOv3训练及测试自己的数据集

1. 环境

Ubuntu 16.04
Python 3.7.0
torch 1.6.0

2. git YOLOv3项目

2.1. git

git clone https://github.com/pjreddie/darknet
cd darknet
make

2.2. 修改Makefile

修改darknet文件夹中的Makefile
修改为：

GPU=1  # 如果使用GPU设置为1，CPU设置为0
CUDNN=0  # 如果使用CUDNN设置为1，否则为0
OPENCV=0  # 如果调用摄像头，还需要设置OPENCV为1，否则为0
OPENMP=0  # 如果使用OPENMP设置为1，否则为0
DEBUG=0 # 如果使用DEBUG设置为1，否则为0