Top 23 Segmentation Open-Source Projects

• albumentations

  28 13,425 8.9 Python

  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

  14 8,844 4.1 Python

  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

• InfluxDB

  www.influxdata.com featured

  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.

  

• jetson-inference

  11 7,349 7.7 C++

  Hello AI World guide to deploying deep-learning inference networks and deep vision primitives with TensorRT and NVIDIA Jetson.

• yolo_tracking

  8 6,110 9.9 Python

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

• nnUNet

  11 5,045 9.2 Python
  

• segmentation_models

  8 4,611 0.0 Python

  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

  1 4,574 0.0 Python

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

• SaaSHub

  www.saashub.com featured

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

  

• unet

  1 4,445 0.0 Jupyter Notebook

  unet for image segmentation

• sam-hq

  6 3,368 8.2 Python

  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)

• Pointnet_Pointnet2_pytorch

  3 3,187 0.0 Python

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

• HRNet-Semantic-Segmentation

  1 3,018 0.0 Python

  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

  2 2,875 0.0 Python

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

• segment-geospatial

  3 2,660 8.6 Python

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

  

• BlenderProc

  15 2,544 8.3 Python

  A procedural Blender pipeline for photorealistic training image generation

  

• UNetPlusPlus

  2 2,173 3.1 Python

  [IEEE TMI] Official Implementation for UNet++

• detrex

  4 1,821 7.2 Python

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

  

• ml-cvnets

  4 1,681 4.8 Python

  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

  11 1,594 6.3 Python

  [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

  1 1,486 8.1 Python

  PyTorch extensions for fast R&D prototyping and Kaggle farming

• vocal-remover

  6 1,394 4.8 Python

  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

  4 1,365 2.8 Python

  OpenMMLab Model Compression Toolbox and Benchmark.

  

• medicaldetectiontoolkit

  2 1,266 0.0 Python

  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

  16 1,237 4.3 Python

  ✂️ 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

• SaaSHub

  www.saashub.com featured

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

  

Segmentation related posts

• Instance segmentation of small objects in grainy drone imagery

  8 projects | /r/computervision | 9 Dec 2023

• 12-Jun-2023

  3 projects | /r/dailyainews | 12 Jun 2023

• Segment Anything in High Quality

  1 project | /r/aigamedev | 10 Jun 2023

• Segment Anything in High Quality

  2 projects | /r/StableDiffusion | 8 Jun 2023

• GitHub - SysCV/sam-hq: Segment Anything in High Quality

  1 project | /r/StableDiffusion | 8 Jun 2023

• Anyone using segment-geospatial?

  1 project | /r/gis | 28 Apr 2023

• Segment Anything & QGIS?

  2 projects | /r/QGIS | 21 Apr 2023
• A note from our sponsor - InfluxDB
  www.influxdata.com | 1 May 2024

  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. Learn more →

Index

Sponsored

SaaSHub - Software Alternatives and Reviews

SaaSHub helps you find the best software and product alternatives

www.saashub.com