高效的Unet-PyTorch
以EfficientNet为编码器的Unet的PyTorch 1.0实现
有用的笔记
由于解码器路径中存在一些舍入问题(不是错误,因此这是一个功能) :smirking_face: ),输入形状应可被32整除。 例如224x224是适合输入图像的尺寸,但225x225不适合。
要求
Python> = 3.6
> = 1.0.0
安装
安装efficientunet-pytorch :
pip install efficientunet-pytorch
用法
1.高效网
例如,说您想要具有5个类的预训练的efficiencynet-b5模型:
from efficientunet import *
model = EfficientNet . from_name ( 'efficientnet-b5' , n_classes = 5 , pretrained = True
1