Python image-segmentation

Open-source Python projects categorized as image-segmentation

Top 21 Python image-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. | | 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? | | 2023-03-29

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

  • CodiumAI

    TestGPT | Generating meaningful tests for busy devs. Get non-trivial tests (and trivial, too!) suggested right inside your IDE, so you can code smart, create more value, and stay confident when you push.

  • PaddleSeg

    Easy-to-use image segmentation library with awesome pre-trained model zoo, supporting wide-range of practical tasks in Semantic Segmentation, Interactive Segmentation, Panoptic Segmentation, Image Matting, 3D Segmentation, etc.

    Project mention: [Medical Segmentation] The all-in-one 3D medical image segmentation toolkit. From data annotation to model deployment, you are welcome to try it all! | | 2022-12-19


  • U-2-Net

    The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."

    Project mention: [Help] Meta's segment anything - How can I make smooth border ? | | 2023-05-14

    Hi :) I am app/web developer and new to AI. Currently, I am making photo app which can segment all the things in image. I've used meta's segment anything. I've got all the masks but the boundary of masks are very bumpy. So I've tried rembg which uses u2net(salient object detection) and pymatting together. Do I have to use pymatting separately after getting segment from segment anything to improve boundary quality of my segmented output ?

  • segmentation_models

    Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

    Project mention: segmentation-models No module Error | | 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'

  • ailia-models

    The collection of pre-trained, state-of-the-art AI models for ailia SDK

    Project mention: 10$ Full Body Tracking! I'm proud to release ToucanTrack (in Beta!). Get decent FBT with the power of 2 PS3 Eye Cameras and AI! | | 2023-02-20

    If you're looking for the differences in terms of how inference is done, I recommend you take a look at MediaPipe's source code. MediaPipe doesn't use raw code, but uses a "graph" instead (eg. pose_landmark_cpu.pbtxt), which can be visualised using MediaPipe Viz. I also used axinc-ai/ailia-models as the base (preprocessing, inference, postprocessing, etc...) which I further built upon (keypoint refinement, roi from keypoints, filtering / smoothing, etc...)

  • 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'. | | 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.


    ONLYOFFICE Docs — document collaboration in your environment. Powerful document editing and collaboration in your app or environment. Ultimate security, API and 30+ ready connectors, SaaS or on-premises

  • PixelLib

    Visit PixelLib's official documentation

    Project mention: YOLOv6: Redefine state-of-the-art for object detection | | 2022-06-29
  • caer

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

  • cleanvision

    Automatically find issues in image datasets and practice data-centric computer vision.

    Project mention: [D] Is accurately estimating image quality even possible? | | 2023-04-22

    Github: Blogpost:

  • Entity

    EntitySeg Toolbox: Towards Open-World and High-Quality Image Segmentation

  • signature_extractor

    A super lightweight image processing algorithm for detection and extraction of overlapped handwritten signatures on scanned documents using OpenCV and scikit-image.

    Project mention: im trying to automate my simpleton job | | 2023-05-03

    It's active field of research. e.g. here or here

  • cloth-segmentation

    This repo contains code and a pre-trained model for clothes segmentation.

    Project mention: Just a reminder that there is a new 'remove background' extension for a1111 | | 2023-03-15

    u2net_cloth_seg (download, source): A pre-trained model for Cloths Parsing from human portrait. Here clothes are parsed into 3 category: Upper body, Lower body and Full body.

  • detectree

    Tree detection from aerial imagery in Python

  • ISP-Guide

    Image Signal Processing (ISP) Guide

  • smriprep

    Structural MRI PREProcessing (sMRIPrep) workflows for NIPreps (NeuroImaging PREProcessing tools)

    Project mention: [D][R] Image pre-processing for quantitative analysis | | 2022-09-30

    I don't know pyradiomics, it looks interesting. From personal experience I can also recommend the library nilearn (developed by scikit-learn core people) and nipype (and impressive interface to all neuroimaging toolboxes out there. Also, I forgot to mention sMRIprep which is fMRIprpe's little sibling but exclusively for anatomical/structural data. Plus, there's MRIQC, that can extract multiple quality parameters (might provide you additional informative features).

  • uformer-pytorch

    Implementation of Uformer, Attention-based Unet, in Pytorch

  • FedCV

    FedCV: An Industrial-grade Federated Learning Framework for Diverse Computer Vision Tasks

  • Mask-RCNN-TF2.7.0-keras2.7.0

    Mask R-CNN for object detection and instance segmentation on Keras==2.7.0 and TensorFlow==2.7.0

  • fashion-segmentation

    A tensorflow model for segmentation of fashion items out of multiple product images

  • Brain-Tumor-Segmentation-And-Classification

    Brain Tumor Segmentation And Classification using artificial intelligence

  • 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.

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-14.

Python image-segmentation related posts


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

Project Stars
1 albumentations 12,044
2 segmentation_models.pytorch 7,339
3 PaddleSeg 7,115
4 U-2-Net 7,112
5 segmentation_models 4,327
6 ailia-models 1,414
7 pytorch-toolbelt 1,394
8 PixelLib 937
9 caer 688
10 cleanvision 609
11 Entity 441
12 signature_extractor 354
13 cloth-segmentation 311
14 detectree 146
15 ISP-Guide 98
16 smriprep 95
17 uformer-pytorch 80
18 FedCV 60
19 Mask-RCNN-TF2.7.0-keras2.7.0 36
20 fashion-segmentation 23
21 Brain-Tumor-Segmentation-And-Classification 12
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.