
一个利用libtorch(PyTorch C++版)进行C++语义分割训练的库-C/C++开发
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
这是一个基于LibTorch框架的C++语义分割项目,旨在为开发者提供高效的语义分割模型训练工具。通过此库,用户可以方便地利用PyTorch强大的功能来优化和加速其C++应用中的图像处理任务。
一个基于LibTorch(Pytorch C++)的C++可训练语义分割库。骨干网络包括ResNet、ResNext。架构有FPN、U-Net、PAN、LinkNet、PSPNet、DeepLab-V3以及DeepLab-V3+。该库的主要特点如下:高级API,仅需一行代码即可创建神经网络;提供7种用于二进制和多类分割的模型架构(包括著名的Unet);所有编码器均具有经过预训练的权重以加快并提高编码效率;与PyTorch CUDA相比,收敛速度提高了35%或更多。
全部评论 (0)
还没有任何评论哟~


