修改C3模块的结构
- 🍨 本文为🔗365天深度学习训练营 中的学习记录博客
- 🍖 原作者:K同学啊 | 接辅导、项目定制
📕本次任务:将yolov5s网络模型中C3模块中的结构按照如下方式修改,并跑通YOLOv5。
如左图,有3个conv模块,需要改为右图,右图只包含2个conv模块。
📕提示:仅需修改./models/common.py
文件
📕步骤:
(1)首先找到coomon.py中定义C3模块的地方;
(2)然后将代码与上图的左图对应起来,观察需要改动的位置,结合结构,更容易看懂代码;
(3)找到之后按照要求修改,并运行train.py,看是否能跑通。
./models/common.py
中,保存的是v5s各个模块的实现,包括基本模块(如autopad、Conv、Bottleneck、BottleneckCSP、C3、SPP、Concat、Expand和Contract)和重要模块(NMS、AutoShape、Detections、Classify)。
一、commom.py文件解析
yolov5-master的文件结构如下:
yolov5-master
|-classify
|-data
|-models (本次学习需要的文件在这里面)
|-hub
|-segment
|