efficientNe笔记

Efficientnet 原文

使用pip安装efficientnet

pip install efficientnet_pytorch

导入efficientnet

from efficientnet_pytorch import EfficientNet
model = EfficientNet.from_name('efficientnet-b0')  # load an EfficientNet
model = EfficientNet.from_pretrained('efficientnet-b0', num_classes=2)  # load a pretrained EfficientNet 

EfficientNet 提取特征

from efficientnet_pytorch import EfficientNet
model = EfficientNet.from_pretrained('efficientnet-b0')

# ... image preprocessing as in the classification example ...
print(img.shape) # torch.Size([1, 3, 224, 224])

features = model.extract_features(img)
print(features.shape) # torch.Size([1, 1280, 7, 7])

网络结构为：

图源自于https://blog.csdn.net/Trent1985/article/details/91126085