Python Segmentation

Open-source Python projects categorized as Segmentation

Top 23 Python Segmentation Projects

  • albumentations

    Fast image augmentation library and an easy-to-use wrapper around other libraries. Documentation: Paper about the library:

    Project mention: Burn Deep Learning Framework Release 0.7.0: Revamped (de)serialization, optimizer & module overhaul, initial ONNX support and tons of new features. | /r/rust | 2023-05-06

    Is something planned to support data augmentations? Something like

  • segmentation_models.pytorch

    Segmentation models with pretrained backbones. PyTorch.

    Project mention: [D] Improvements/alternatives to U-net for medical images segmentation? | /r/MachineLearning | 2023-03-29

    SMP offers a wide variety of segmentation models with the option to use pre-trained weights.

  • InfluxDB

    Access the most powerful time series database as a service. Ingest, store, & analyze all types of time series data in a fully-managed, purpose-built database. Keep data forever with low-cost storage and superior data compression.

  • segmentation_models

    Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

    Project mention: segmentation-models No module Error | /r/computervision | 2022-10-04

    I used segmentation-models ( to create a deeplabv3+ model. I havent used it in the last 2 months and now i comeback to the same code and cant use it. Getting ModuleNotFoundError: No module named 'segmentation_models_pytorch.deeplabv3'

  • pointnet

    PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation

    Project mention: How do I balance a pointcloud dataset without compromising it? | /r/learnmachinelearning | 2022-10-25

    I am using pointnet, A neural network which directly processes point clouds and labels them on a per-point basis.

  • nnUNet

    Project mention: Where to even begin to extract yellow spots from these pictures? | /r/MLQuestions | 2023-04-02

    Now just run your images and masks in nnUNet to see if it works.

  • HRNet-Semantic-Segmentation

    The OCR approach is rephrased as Segmentation Transformer: This is an official implementation of semantic segmentation for HRNet.

    Project mention: Some general and HRNetV2+OCR related questions (Semantic Segmentation) | /r/computervision | 2022-11-30

    Hi everyone. I'm trying utilize popular HRNetV2+OCR model posted in this repo

  • pointnet2

    PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Space

    Project mention: Why does trained model fail on new data? | /r/learnmachinelearning | 2022-08-26

    Is this how you trained the segmentation?

  • Sonar

    Write Clean Python Code. Always.. Sonar helps you commit clean code every time. With over 225 unique rules to find Python bugs, code smells & vulnerabilities, Sonar finds the issues while you focus on the work.

  • Pointnet_Pointnet2_pytorch

    PointNet and PointNet++ implemented by pytorch (pure python) and on ModelNet, ShapeNet and S3DIS.

  • BlenderProc

    A procedural Blender pipeline for photorealistic training image generation

    Project mention: Synthetic image Generation | /r/learnmachinelearning | 2023-03-22

    Blender with add-ons (Kubric, BlenderProc)

  • segment-geospatial

    A Python package for segmenting geospatial data with the Segment Anything Model (SAM)

    Project mention: Anyone using segment-geospatial? | /r/gis | 2023-04-28

    Maybe try reporting as a GitHub issue?

  • pytorch-toolbelt

    PyTorch extensions for fast R&D prototyping and Kaggle farming

    Project mention: Surrender cards are being distributed to Russianforces: 'Your ticket to a peaceful life. Show this card to a Ukrainian soldier - it will save your life and help you get back home' On the back: a telegram chat & phone number they can contact 'to receive detailed support'. | /r/ukraine | 2022-09-12

    For what it's worth, the QR code points to a URI shortener domain "" which redirects to a cloudflare hosted page that contains telegram information relating to "nikolay_bodenko" and "chaos_admin" ("For advertising and cooperation"? Whatever that is supposed to mean), which led me to "rf200_now" and "rf200_nooow" (looks to get spammed around a bit), which is purportedly supportive of Ukraine.

  • ml-cvnets

    CVNets: A library for training computer vision networks

  • medicaldetectiontoolkit

    The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.

  • XMem

    [ECCV 2022] XMem: Long-Term Video Object Segmentation with an Atkinson-Shiffrin Memory Model

    Project mention: Track-Anything: a flexible and interactive tool for video object tracking and segmentation, based on Segment Anything and XMem. | /r/StableDiffusion | 2023-04-25

    Nvm just found the occlusion video on holy shit

  • PaddleViT

    :robot: PaddleViT: State-of-the-art Visual Transformer and MLP Models for PaddlePaddle 2.0+

  • mmrazor

    OpenMMLab Model Compression Toolbox and Benchmark.

    Project mention: MMDeploy: Deploy All the Algorithms of OpenMMLab | /r/u_Allent_pjlab | 2022-11-21

    MMRazor: OpenMMLab model compression toolbox and benchmark.

  • PixelLib

    Visit PixelLib's official documentation

    Project mention: YOLOv6: Redefine state-of-the-art for object detection | | 2022-06-29
  • image-background-remove-tool

    ✂️ Automated high-quality background removal framework for an image using neural networks. ✂️

    Project mention: Show HN: Image background removal without annoying subscriptions | | 2023-05-25
  • vocal-remover

    Vocal Remover using Deep Neural Networks

    Project mention: Dúvida - App/Site para retirar a voz de uma música | /r/portugal | 2023-05-19
  • ttach

    Image Test Time Augmentation with PyTorch!

  • caer

    High-performance Vision library in Python. Scale your research, not boilerplate.

  • segmenter

    [ICCV2021] Official PyTorch implementation of Segmenter: Transformer for Semantic Segmentation

  • segment-anything-video

    MetaSeg: Packaged version of the Segment Anything repository

    Project mention: Sharing an automatic tool I built for video background removal (free and better than runwayml) | /r/StableDiffusion | 2023-05-04
  • SaaSHub

    SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020). The latest post mention was on 2023-05-25.

Python Segmentation related posts


What are some of the best open-source Segmentation projects in Python? This list will help you:

Project Stars
1 albumentations 12,072
2 segmentation_models.pytorch 7,399
3 segmentation_models 4,332
4 pointnet 4,198
5 nnUNet 3,724
6 HRNet-Semantic-Segmentation 2,873
7 pointnet2 2,666
8 Pointnet_Pointnet2_pytorch 2,404
9 BlenderProc 2,045
10 segment-geospatial 1,611
11 pytorch-toolbelt 1,395
12 ml-cvnets 1,377
13 medicaldetectiontoolkit 1,203
14 XMem 1,187
15 PaddleViT 1,116
16 mmrazor 1,073
17 PixelLib 939
18 image-background-remove-tool 921
19 vocal-remover 911
20 ttach 841
21 caer 691
22 segmenter 686
23 segment-anything-video 649
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives