文件名称:YOLOV4:You Only Look Once目标检测模型在pytorch当中的实现-Python开发
文件大小:5.32MB
文件格式:ZIP
更新时间:2024-06-14 10:04:28
Python Deep Learning
这是一个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的图片的。
【文件预览】:
yolov4-pytorch-master
----常见问题汇总.md(20KB)
----nets()
--------CSPdarknet.py(7KB)
--------yolo_training.py(23KB)
--------yolo4.py(7KB)
----model_data()
--------yolo_anchors.txt(85B)
--------simhei.ttf(9.3MB)
--------coco_classes.txt(705B)
--------voc_classes.txt(153B)
----yolo.py(16KB)
----get_gt_txt.py(3KB)
----get_dr_txt.py(7KB)
----LICENSE(1KB)
----kmeans_for_anchors.py(4KB)
----get_map.py(33KB)
----utils()
--------dataloader.py(10KB)
--------utils.py(15KB)
----README.md(6KB)
----test.py(566B)
----VOCdevkit()
--------VOC2007()
----voc_annotation.py(2KB)
----predict.py(5KB)
----img()
--------street.jpg(437KB)
----logs()
--------README.md(30B)
----train.py(15KB)