轻量化骨干系列(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):