[fy目标检测] Yolov5-model.py源码详解

  • A+
所属分类:专业基础

在yolov5源码中,models文件下

1
2
class DetectionModel(BaseModel):
def __init__(self, cfg='yolov5s.yaml', ch=3, nc=None, anchors=None):

默认使用的是yolov5s.yanl,我们可以在同目录下找到这个文件

 

[fy目标检测] Yolov5-model.py源码详解

整体的架构如下

 

[fy目标检测] Yolov5-model.py源码详解

 

整体架构图展示了每一层之间的联系

 

[fy目标检测] Yolov5-model.py源码详解

 

可以把backbone当成一个提取特征的一个模块,低层更适合预测小目标,中间的模块相当于做了一个特征融合,最终会在三个层级上做目标检测。也就是源码中第 17 20,23层。

anchors:预定义好的一些矩形框,8,16,32分别对应低中高三个层级。以第一行为例,他就是定义了三个anchors:10*13,16*30,33*23。

Detect是整个流程的最后一个模块。

weinxin
XCJinggai的微信
这是我的微信二维码,有任何问题请加微信联系我!或发送邮件至lqzhangfengxi@163.com
zfy

发表评论

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen: