Shypd.aiThe largest open index of AI toolssegmentation_models.pytorch is an Open Source library for semantic image segmentation based on PyTorch. It offers 12 encoder-decoder architectures and over 800 pretrained convolutional and transformer-based backbones, simplifying neural network creation to just two lines of code.
Trending
About
segmentation_models.pytorch is an Open Source Python library designed for semantic image segmentation using PyTorch. It provides a high-level API that allows users to create neural networks with minimal code, supporting 12 encoder-decoder model architectures such as Unet, Unet++, Segformer, and DPT. The library boasts an extensive collection of over 800 pretrained convolutional and transformer-based encoders, including timm support, which helps achieve faster and more stable convergence during training. It also includes popular metrics and losses for training routines, such as Dice and Jaccard, and is compatible with ONNX export and torch script/trace/compile. This makes it a versatile tool for researchers and practitioners in computer vision.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
