segmentation-models-pytorch · PyPI
pypi.org › project › segmentation-models-pytorchNov 18, 2021 · Segmentation based on PyTorch. The main features of this library are: High level API (just two lines to create a neural network) 9 models architectures for binary and multi class segmentation (including legendary Unet) 113 available encoders. All encoders have pre-trained weights for faster and better convergence.