【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: .


最近有几个想法

  1. 想在yolov5的基础上加一个分支进行分割任务
  2. yolov5 + Circular Smooth Label -> 旋转目标检测
  3. yolov5剪枝轻量化

已实现:

  1. YOLOv5 Head解耦:
  2. yolov5+shufflenetv2轻量化

一、导航

YOLOV5-U

    data models utils

二、尚存在的问题

  1. AWS Inferentia?
  2. non_max_suppression中的autolabelling是什么?这样子作不就导致最终mAP过高了吗?为什么这么做? 更新:已解决,在general.py的non_max_suppression函数中有详细解释!!!
  3. 计算损失函数置信度损失这一步时为什么要对置信度进行排序???排不排序效果不是一样的吗? 更新:已解决,在loss.py的ComputeLoss类中已经解释清楚!

Reference

  1. Github: .
  2. Github: .
  3. BiliBili: .
  4. BiliBili 霹雳吧啦Wz: .
  5. BiliBili 人工智能打游戏系列课程1:基于深度学习的目标检测算法: .
  6. Laughing-q: .
  7. 幻灵H_Ling: .
  8. 昌山小屋: .
  9. 恩泽君: .
  10. TheOldManAndTheSea: .
  11. 菊头蝙蝠: .
  12. 外交官的后花园: .
  13. guikunchen: .
  14. 迷途小书童的Note: .
  15. RainbowSun1102: .
  16. Activewaste: .
  17. DEEPFELLOW: .
  18. cv君: .
  19. aabbcccddd01: .
  20. 技术挖掘者: .
  21. 那年当上博士前: .
  22. 啥都会一点的老程: .
  23. 一只歪脖子程序猿: .
  24. 是否龙磊磊真的一无所有: .
  25. YY_172: .
  26. 菊头蝙蝠: .
  27. 暮丶凉: .
  28. wa1tzy: .
  29. DLANDML: .
  30. 昌山小屋: .
  31. Liaojiajia-2020: .
  32. Liaojiajia-2020: .
  33. Liaojiajia-2020: .
  34. TheOldManAndTheSea: .
  35. 吸欧大王: .
  36. 昌山小屋: .
  37. BIT可达鸭: .
  38. xingzeng307: .
  39. DEEPFELLOW: .
  40. Charles.zhang: .
  41. a_cheng_: .
  42. Liaojiajia-2020: .
  43. lambda: .
经验分享 程序员 微信小程序 职场和发展