【文件属性】：
文件名称：EfficientUnet-PyTorch:以EfficientNet为编码器的Unet的PyTorch 1.0实现
文件大小：11KB
文件格式：ZIP
更新时间：2021-05-05 22:39:56
image-segmentation unet unet-pytorch efficientnet efficientunet
高效的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
【文件预览】：
EfficientUnet-PyTorch-master
----efficientunet()
--------efficientnet.py(8KB)
--------_version.py(22B)
--------utils.py(7KB)
--------__init__.py(91B)
--------layers.py(8KB)
--------efficientunet.py(7KB)
----main.py(609B)
----LICENSE(1KB)
----setup.py(906B)
----README.md(2KB)
----.gitignore(23B)