A c++ trainable semantic segmentation library based on libtorch (pytorch c++). Backbone: VGG, ResNet, ResNext. Architecture: FPN, U-Net, PAN, LinkNet, PSPNet, DeepLab-V3, DeepLab-V3+ by now.
neural-network cpp models pytorch imagenet resnet image-segmentation unet semantic-segmentation resnext pretrained-weights pspnet fpn deeplabv3 deeplabv3plus libtorch pytorch-cpp pytorch-cpp-frontend pretrained-backbones libtorch-segment
-
Updated
Dec 7, 2023 - C++