SEnet

论文翻译:https://blog.csdn.net/Quincuntial/article/details/78605507 原理: 类似于之前的一篇文章, 自动学习每个channel 的权重。 from torch import nn class SELayer(nn.Module): def __init__(self, channel, reduction=16):
相关文章
相关标签/搜索