Segmentation

Open-source projects categorized as Segmentation

Top 23 Segmentation Open-Source 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

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

  • jetson-inference

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

  • yolo_tracking

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

    Project mention: FLiPN-FLaNK Stack Weekly for 17 April 2023 | dev.to | 2023-04-17
  • 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. https://github.com/MIC-DKFZ/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

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

  • unet

    unet for image 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)

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

Segmentation related posts

Index

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

Project Stars
1 albumentations 13,272
2 segmentation_models.pytorch 8,700
3 jetson-inference 7,235
4 yolo_tracking 5,930
5 nnUNet 4,863
6 segmentation_models 4,583
7 pointnet 4,529
8 unet 4,408
9 sam-hq 3,239
10 Pointnet_Pointnet2_pytorch 3,108
11 HRNet-Semantic-Segmentation 3,018
12 pointnet2 2,875
13 segment-geospatial 2,610
14 BlenderProc 2,496
15 UNetPlusPlus 2,138
16 detrex 1,778
17 ml-cvnets 1,653
18 XMem 1,568
19 pytorch-toolbelt 1,474
20 vocal-remover 1,344
21 mmrazor 1,335
22 medicaldetectiontoolkit 1,262
23 image-background-remove-tool 1,190
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives
www.saashub.com