
一个采用libtorch (PyTorch C++) 的C++可训练语义分割库 - C/C++开发
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
这是一个基于LibTorch(PyTorch C++绑定)构建的开源C++库,专注于提供高效、灵活的语义分割模型训练功能,适合于需要深度学习能力但偏好使用C++语言的开发者和研究人员。
这是一个基于LibTorch(Pytorch C++)的C++可训练语义分割库。骨干网络包括ResNet和ResNext,架构则有FPN、U-Net、PAN、LinkNet、PSPNet、DeepLab-V3及DeepLab-V3+等多种选择。
该库的主要特点如下:
1. 高级API:仅需一行代码即可创建神经网络。
2. 支持七种用于二元和多类分割的模型架构,包括著名的U-Net。
3. 提供多种编码器选项,并且所有编码器均配备了预训练权重以实现更快、更好的性能。
4. 相较于PyTorch CUDA版本,收敛速度提高了至少35%。
全部评论 (0)
还没有任何评论哟~


