-
segmentation_models.pytorch:具有预训练骨干的细分模型。 火炬下载
资源介绍
具有用于图像的神经网络的Python库基于分割。
该库的主要功能是：
高级API（只需两行即可创建神经网络）
9种用于二进制和多类细分的模型架构（包括传奇的Unet）
104种可用编码器
所有编码器均具有预训练的权重，以实现更快更好的收敛
请访问或阅读以下自述文件，以了解有关细分模型Pytorch（简称SMP）库的更多信息
:clipboard: 表中的内容
建筑学
编码器
型号API
输入通道
辅助分类输出
深度
安装
图书馆比赛获胜
贡献
引用
执照
:hourglass_not_done: 快速开始
1.使用SMP创建第一个细分模型
细分模型只是一个PyTorch nn.Module，可以轻松创建：
import segmentation_models_pytorch as smp
model = smp . Unet (
encoder_name = "resnet34" , # choos