文件名称：EfficientUnet-PyTorch:以EfficientNet为编码器的Unet的PyTorch 1.0实现
文件大小：11KB
文件格式：ZIP
更新时间：2024-05-25 16:26:36
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)