使用yolo-v3和yolov3-tiny训练自己的数据
yolov3的官网:教程官网都有,老详细了。
确立我们的目标:
训练我们自己的数据集,从而生成一个XXX.weights文件,这就是炼丹练出来的丹
先搞定darknet
安装+编译
- 先建个文件夹
- 打开命令行进入这个文件夹
git clone https://github.com/pjreddie/darknet建议翻墙下,有点小大cd darknetmake- ok darknet算是成功安装了
测试一下yolo能不能用
- 在
darknet文件夹中 - 下载这个放到
darknet文件夹中(不翻墙的速度感人)yolov3.weights - 运行命令
./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg - 应该就能看到结果了,在darknet 文件夹下生成了
predictions.jpg图片
我们在试试tiny-yolo的检测
- 下载这个放到darknet文件夹中(不翻墙的速度感人)yolov3-tiny.weights
- 结果同上,不做赘述了
训练的数据准备
-
准备好训练图片,保存在
train_imgs -
准备好测试图片,保存在
test_imgs -
准备好训练集
- 建一个文件夹
train_dataset - 里面含有
train_imags的全部图片,比如img1.jpg,img2.jpg,img3.jpg…… - 同含有每张图片对应的标注信息,比如
img1.txt,img2.txt,img3.txt……,也就是说每有一张图片,就有一个图片名.txt的文件与之对应。 图片名.txt里的内容是
label名 x y w h
label名 x y w h
label名 x y w h
label名 x y w h
……
x y w h 计算法则如下:
x = X ( 框 中 心 x 坐 标 ) W ( 图 宽 ) x=\frac{X(框中心x坐标)}{W(图宽)} x=W(图宽)
- 建一个文件夹

本文详细介绍如何使用 Yolo-v3 和 yolov3-tiny 训练自定义数据集,涵盖从数据准备、标注到模型训练的全过程。通过实例演示如何设置配置文件、运行训练命令及调整模型参数。
最低0.47元/天 解锁文章
3万+





