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: https://albumentations.ai/docs/ Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

    Project mention: Augment specific classes? | /r/computervision | 2023-12-06

    You can use albumentations if you are comfortable with using open source libraries https://github.com/albumentations-team/albumentations

  • segmentation_models.pytorch

    Segmentation models with pretrained backbones. PyTorch.

    Project mention: Instance segmentation of small objects in grainy drone imagery | /r/computervision | 2023-12-09

    Also, I’d suggest considering switching to the segmentation-models library - it provides U-Net models with a variety of pretrained backbones of as encoders. The author also put out a PyTorch version. https://github.com/qubvel/segmentation_models.pytorch https://github.com/qubvel/segmentation_models

  • WorkOS

    The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.

  • yolo_tracking

    BoxMOT: pluggable SOTA tracking modules for segmentation, object detection and pose estimation models

  • nnUNet

  • segmentation_models

    Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

    Project mention: Instance segmentation of small objects in grainy drone imagery | /r/computervision | 2023-12-09

    Also, I’d suggest considering switching to the segmentation-models library - it provides U-Net models with a variety of pretrained backbones of as encoders. The author also put out a PyTorch version. https://github.com/qubvel/segmentation_models.pytorch https://github.com/qubvel/segmentation_models

  • pointnet

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

  • sam-hq

    Segment Anything in High Quality [NeurIPS 2023]

    Project mention: 12-Jun-2023 | /r/dailyainews | 2023-06-12

    Segment Anything in High Quality (https://arxiv.org/abs/2306.01567)

  • InfluxDB

    Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

  • Pointnet_Pointnet2_pytorch

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

  • HRNet-Semantic-Segmentation

    The OCR approach is rephrased as Segmentation Transformer: https://arxiv.org/abs/1909.11065. This is an official implementation of semantic segmentation for HRNet. https://arxiv.org/abs/1908.07919

  • pointnet2

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

  • 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? https://github.com/opengeos/segment-geospatial/issues

  • BlenderProc

    A procedural Blender pipeline for photorealistic training image generation

  • UNetPlusPlus

    [IEEE TMI] Official Implementation for UNet++

  • detrex

    detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.

  • ml-cvnets

    CVNets: A library for training computer vision networks

    Project mention: Apple Researchers Introduce ByteFormer: An AI Model That Consumes Only Bytes And Does Not Explicitly Model The Input Modality - MarkTechPost | /r/singularity | 2023-06-16

    https://github.com/apple/ml-cvnets/tree/main/examples/byteformer - Where the code will be located once uploaded

  • XMem

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

    Project mention: [D] Which open source models can replicate wonder dynamics's drag'n'drop cg characters? | /r/MachineLearning | 2023-06-11

    Use Segmentation Model (SAM) combined with Inpainting model (E2FGVI) and Xmem to cut out the live action subject.

  • pytorch-toolbelt

    PyTorch extensions for fast R&D prototyping and Kaggle farming

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

    OpenMMLab Model Compression Toolbox and Benchmark.

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

  • 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 | news.ycombinator.com | 2023-05-25
  • PaddleViT

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

  • PixelLib

    Visit PixelLib's official documentation https://pixellib.readthedocs.io/en/latest/

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

Python Segmentation related posts

Index

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

Project Stars
1 albumentations 13,362
2 segmentation_models.pytorch 8,773
3 yolo_tracking 6,046
4 nnUNet 4,960
5 segmentation_models 4,597
6 pointnet 4,556
7 sam-hq 3,324
8 Pointnet_Pointnet2_pytorch 3,150
9 HRNet-Semantic-Segmentation 3,018
10 pointnet2 2,875
11 segment-geospatial 2,647
12 BlenderProc 2,516
13 UNetPlusPlus 2,162
14 detrex 1,807
15 ml-cvnets 1,662
16 XMem 1,584
17 pytorch-toolbelt 1,479
18 vocal-remover 1,377
19 mmrazor 1,352
20 medicaldetectiontoolkit 1,265
21 image-background-remove-tool 1,220
22 PaddleViT 1,169
23 PixelLib 1,008
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives
www.saashub.com