1、前言
之前介绍了yolov5如何更换C2f模块以及加入注意力机制SE模块的示例,详细请参考本专栏:
YOLOV5 实战项目(训练、部署、改进等等)_听风吹等浪起的博客-CSDN博客
本文将详细介绍yolov5更换官方backbone,以轻量级网络mobilenet为例。因为mobilenet是轻量级的小型网络,参数量和计算量相比于传统的DarkNet而言少了很多,可能精度会有所损失。不过掌握了方法后,可以熟练的更换别的网络,例如transformer等等
2、MobileNet 网络
因为要引入MobileNet网络,这里简单做个介绍,详细可以参考分类部分文章:
MobileNet V1 图像分类_mobilenet 图像-CSDN博客
为了方便,这里我们通过torchvision 官方实现的mobilenet网络
这里的pretrained 是指载入官方的预训练权重,progress第一次从官网下载是否打印进度条(可忽略)
标签:YOLOV5,yolov5,MobileNet,mobilenet,为例,网络,博客,轻量级 From: https://blog.csdn.net/qq_44886601/article/details/136838903