上传者: 42129797
|
上传时间: 2021-12-15 13:27:50
|
文件大小: 6.48MB
|
文件类型: -
一个基于libtorch(pytorch c ++)的c ++可训练语义分割库。
骨干网:ResNet,ResNext。
架构:到目前为止,FPN,U-Net,PAN,LinkNet,PSPNet,DeepLab-V3,DeepLab-V3 +。
英文|
基于LibTorch的带有神经网络的C ++库,用于图像分割。
:star:如果这个项目对您有帮助,请给个星。
:star:这个库的主要特点是:高级API(仅是创建神经网络的一行)7种用于二进制和多类分割的模型架构(包括传奇的Unet)7种可用的编码器所有编码器均具有经过预先训练的权重,以实现更快,更好的编码与pytorch cuda相比,收敛速度提高了35%或更多,