yolov5模型压缩-mobilenetv3

最新推荐文章于 2024-06-13 17:11:45 发布

不想学习！！！

最新推荐文章于 2024-06-13 17:11:45 发布

阅读量144

点赞数

分类专栏：模型压缩文章标签： YOLO 轻量化模型压缩

本文链接：https://blog.csdn.net/weixin_43864650/article/details/137599258

版权

模型压缩专栏收录该内容

7 篇文章 7 订阅 ¥19.90 ¥99.00

订阅专栏

轻量化骨干系列（2）–mobilenetv3
参考论文：https://openaccess.thecvf.com/content_ICCV_2019/papers/Howard_Searching_for_MobileNetV3_ICCV_2019_paper.pdf
yolov5改进流程：
1、在common.py中添加如下模块：

from torchvision import models
import torch
import torch.nn as nn

class SimAM(nn.Module):
    def __init__(self, lamda=1e-5):
        super().__init__()
        self.lamda