EfficientNet-Model(pytorch版本)
《》
作者:
单位:
发表会议及时间:
EfficientNet从网络的宽度、深度和分辨率三个角度进行组合缩放,来寻找最优网络
一 论文导读
重新审视了前人提出的模型放缩的几个维度:网络深度、网络宽度、图像分辨率,前人的文章多是放大其中的一个维度以达到更高的准确率,比如 ResNet-18 到 ResNet-152 是通过增加网络深度的方法来提高准确率。
本文跳出了前人对放缩模型的理解,从一个高度去审视这些放缩维度。认为这三个维度之间是互相影响的并探索出了三者之间最好的组合,在此基础上提出了最新的网络 EfficientNet。
二 论文精读
三 代码实现
https://github.com/lukemelas/EfficientNet-PyTorch
model = EfficientNet.from_pretrained('efficientnet-b1', num_classes=23)