使用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