【Week Y5】yolo.py文件解读,插入C2模块到指定位置

本文介绍了如何在yolov5s项目中插入C2模块,包括在common.py中的模块定义、yolo.py中parse_model的修改、yolov5s.yaml的backbone配置调整,以及执行训练过程和验证网络结构的步骤。
摘要由CSDN通过智能技术生成

在这里插入图片描述
模块结构如下:【同Y4】
在这里插入图片描述
【Y4】周直接在C3的模块中减少了一次Conv,本次需要还原C3模块,并重新定义一个C2模块。

C2模块插入位置如下:
在这里插入图片描述

一、common.py文件修改

在原C3模块附近,增加C2模块定义:
在这里插入图片描述

二、yolo.py文件修改

找到parse_model()模块,按如下修改:
在这里插入图片描述

parse_model()模块用于将模型的各个模块拼接起来,从而形成一个完整的网络结构。

三、yolov5s.yaml修改

找到backbone位置,按如下修改:
在这里插入图片描述

四、训练

执行:python .\train.py --img 900 --batch 2 --epoch 100 --data .\fruit.yaml --cfg .\models\yolov5s.yaml --weights .\yolov5s.pt --device cpu
在这里插入图片描述

打印的网络结构如下:
在这里插入图片描述
训练过程正常:
在这里插入图片描述

  • 9
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值