0
点赞
收藏
分享

微信扫一扫

MobileNetV3轻量化YOLOv8

Java旺 2024-06-23 阅读 54

1 轻量化模型

一般而言,模型轻量化有三个途径:
知识蒸馏:大模型引导小模型训练,让其逼近大模型效果
轻量化模块替换:利用一些轻量化模块进行替换,减少模型参数
剪枝:通过优化算法引导模型裁剪无用的参数

MobileNetV3论文如下,自行搜索
在这里插入图片描述

2 修改步骤

  • nn/modules的文件夹下新建一个backbone文件夹,在backbone文件夹里面新建一个MobileNetV3.py文件,放入MobileNetV3源码。

MobileNetV3.py如下所示

from torch import nn


# ######  Mobilenetv3
class h_sigmoid(nn
举报

相关推荐

0 条评论