【YOLOV5-5.x 源码讲解】整体项目文件导航
前言
这个项目是github的开源项目,YOLOV5:,目前已经有14.1k个Star 和 4.9k 个Folk了,非常的火。下面我会给大家逐个的文件介绍这个项目中的所有代码,希望能帮到大家。
原本我下的2021年4月12日更新的v5.0版本,整个项目我做了一点点的文件位置的改变,也加了点自己在其他论文中学到的tricks。不过由于yolov5这个版本迭代的速度非常快,我这个代码也并不是和官网下载v5.0的那个版本就一模一样,而且我也参考了一些后面的版本代码,还是稍微改动了一些的,不过总体来说还是差不了太多的。下面我会按文件逐个函数的进行攻破,因为我本人也还是研一,肯定会有很多的不足,希望大家指正!!!
现在是2021-07-23,目标是在一个月内写完,也就是最迟2021-8-23日更完整个项目的所有的代码。
拖几天,还有3个文件!!!
写完了,断断续续写了35天,主要是最近比较忙,好在没拖到开学。希望对大家有些帮助!
注释版全部项目文件已上传至GitHub: .
最近有几个想法
- 想在yolov5的基础上加一个分支进行分割任务
- yolov5 + Circular Smooth Label -> 旋转目标检测
- yolov5剪枝轻量化
已实现:
- YOLOv5 Head解耦:
- yolov5+shufflenetv2轻量化
一、导航
YOLOV5-U
-
data models utils
二、尚存在的问题
- AWS Inferentia?
- non_max_suppression中的autolabelling是什么?这样子作不就导致最终mAP过高了吗?为什么这么做? 更新:已解决,在general.py的non_max_suppression函数中有详细解释!!!
- 计算损失函数置信度损失这一步时为什么要对置信度进行排序???排不排序效果不是一样的吗? 更新:已解决,在loss.py的ComputeLoss类中已经解释清楚!
Reference
- Github: .
- Github: .
- BiliBili: .
- BiliBili 霹雳吧啦Wz: .
- BiliBili 人工智能打游戏系列课程1:基于深度学习的目标检测算法: .
- Laughing-q: .
- 幻灵H_Ling: .
- 昌山小屋: .
- 恩泽君: .
- TheOldManAndTheSea: .
- 菊头蝙蝠: .
- 外交官的后花园: .
- guikunchen: .
- 迷途小书童的Note: .
- RainbowSun1102: .
- Activewaste: .
- DEEPFELLOW: .
- cv君: .
- aabbcccddd01: .
- 技术挖掘者: .
- 那年当上博士前: .
- 啥都会一点的老程: .
- 一只歪脖子程序猿: .
- 是否龙磊磊真的一无所有: .
- YY_172: .
- 菊头蝙蝠: .
- 暮丶凉: .
- wa1tzy: .
- DLANDML: .
- 昌山小屋: .
- Liaojiajia-2020: .
- Liaojiajia-2020: .
- Liaojiajia-2020: .
- TheOldManAndTheSea: .
- 吸欧大王: .
- 昌山小屋: .
- BIT可达鸭: .
- xingzeng307: .
- DEEPFELLOW: .
- Charles.zhang: .
- a_cheng_: .
- Liaojiajia-2020: .
- lambda: .
上一篇:
IDEA上Java项目控制台中文乱码