EfficientNet的pyTorch版本的使用和训练方法

EfficientNet的pytorch版本的训练和评估使用方法

1. EfficientNet 的pyTorch版本的测试和使用

第三方PyTorch代码

# pytorch 的efficientNet安装
Install via pip:
pip install efficientnet_pytorch

Or install from source:
git clone https://github.com/lukemelas/EfficientNet-PyTorch
cd EfficientNet-Pytorch
pip install -e

# 使用示例
Usage

Loading pretrained models

Load an EfficientNet:
from efficientnet_pytorch import EfficientNet
model = EfficientNet.from_name('efficientnet-b0')

Load a pretrained EfficientNet:
from efficientnet_pytorch import EfficientNet
model = EfficientNet.from_pretrained('efficientnet-b0')

#修改输出通道数由1000改为50示例
from efficientnet_pytorch import EfficientNet
from torch import nn
model = EfficientNet.from_pretrained('efficientnet-b5')
feature = model._fc.in_features
model._fc = nn.Linear(in_features=feature,out_features=50,bias=True)
print(model)

2.提取特征使用

提取特征时可以使用model.extract_features来实现
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