
EfficientUnet-PyTorch:基于EfficientNet编码器的Unet的PyTorch 1.0版本实现(包含源码)。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
高效的Unet-PyTorch,利用EfficientNet作为编码器的Unet实现,在PyTorch 1.0版本中提供了诸多有价值的说明。值得注意的是,由于解码器路径中存在一些舍入特性(并非错误,因此它是一个设计上的功能),输入图像的形状必须能够被32整除。例如,224x224尺寸的图像是理想的输入尺寸,但225x225尺寸的图像则不适用。该实现要求Python版本大于等于3.6以及PyTorch 1.0.0。安装efficientunet-pytorch 的步骤如下:使用 pip 安装:pip install efficientunet-pytorch。
使用方法如下:首先,如果您希望使用具有5个类且预训练的EfficientNet-b5模型的Unet,可以这样操作:从 efficientunet 模块导入相关代码,然后创建并初始化 EfficientNet 模型:`from efficientunet import * model = EfficientNet.from_name(‘efficientnet-b5’, n_classes=5, pretrained=True`。
全部评论 (0)
还没有任何评论哟~


