这是一个YoloV4-pytorch的源码,可以用于训练自己的模型。
YOLOV4:You Only Look Once目标检测模型在pytorch当中的实现
目录
实现的内容 Achievement
所需环境 Environment
注意事项 Attention
小技巧的设置 TricksSet
文件下载 Download
预测步骤 How2predict
训练步骤 How2train
参考资料 Reference
实现的内容
主干特征提取网络:DarkNet53 => CSPDarkNet53
特征金字塔:SPP,PAN
训练用到的小技巧:Mosaic数据增强、Label Smoothing平滑、CIOU、学习率余弦退火衰减
激活函数:使用Mish激活函数
……balabla
所需环境
torch==1.2.0
注意事项
代码中的yolo4_weights.pth是基于608x608的图片训练的,但是由于显存原因。我将代码中的图片大小修改成了416x416。有需要的可以修改回来。 代码中的默认anchors是基于608x608的图片的。
1