segmentation_models.pytorch
时间: 2023-04-24 15:03:16 浏览: 213
Python库 | segmentation_models_pytorch-0.0.2.tar.gz
5星 · 资源好评率100%
segmentation_models.pytorch是一个基于PyTorch框架的语义分割模型库,提供了多种经典的语义分割模型,如UNet、FPN、LinkNet等,并且支持多种预训练模型权重,如ResNet、ResNeXt、MobileNet等。该库还提供了方便的训练和评估工具,可以帮助用户快速构建和训练自己的语义分割模型。
阅读全文