{"repo":"qubvel-org/segmentation_models.pytorch","free":true,"listed":false,"github":"https://github.com/qubvel-org/segmentation_models.pytorch","clone":"git clone https://github.com/qubvel-org/segmentation_models.pytorch.git","description":"Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.","language":"Python","stars":11697,"topics":["segmentation","image-processing","pspnet","unet","unet-pytorch","pytorch","fpn","models","imagenet","semantic-segmentation"],"license":"MIT","category":"machine-learning","readme_excerpt":"Python library with Neural Networks for Image Semantic Segmentation based on PyTorch. The main features of the library are: - Super simple high-level API (just two lines to create a neural network) - 12 encoder-decoder model architectures (Unet, Unet++, Segformer, DPT, ...) - 800+ pretrained convolution- and transform-based encoders, including timm support - Popular metrics and losses for training routines (Dice, Jaccard, Tversky, ...) - ONNX export and torch script/trace/compile friendly 🤝 Sponsor: withoutBG withoutBG is a high-quality background removal tool. They built their open-source image matting and refiner models using smp.Unet and are proudly sponsoring this project. 📚 Project Documentation 📚 Visit Read The Docs Project Page or read the following README to know more about Segmentation Models Pytorch (SMP for short) library 📋 Table of content 1. Quick start 2. Examples 3. Models and encoders 4. Models API 1. Input channels 2. Auxiliary classification output 3. Depth 5. Installation 6. Competitions won with the library 7. Contributing 8. Citing 9. License ⏳ Quick start 1. Create your first Segmentation model with SMP The segmentation model is just a PyTorch torch.nn.Module , which can be created as easy as: - see table with available model architectures - see table with available encoders and their corresponding weights 2. Configure data preprocessing All encoders have pretrained weights. Preparing your data the same way as during weights pre-training may give you","default_branch":null,"files":null,"tree":[],"storefront":"/r/qubvel-org","claimed":false,"request_supported":{"post":"https://gitbuyer.com/r/qubvel-org/segmentation_models.pytorch/request-supported","requests":0},"note":"indexed from public GitHub; nothing is for sale on this page. Clone it from GitHub. Paid listings live at /search."}