Python Segmentation

Open-source Python projects categorized as Segmentation | Edit details

Top 23 Python Segmentation Projects

  • GitHub repo albumentations

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

    Project mention: Data augmentation strategies for object detection? Could you point me to good resources or best practices you know of? | | 2021-10-30

    You can definitely look at Albumentation - we had a ton of success working with this library

  • GitHub repo segmentation_models.pytorch

    Segmentation models with pretrained backbones. PyTorch.

    Project mention: Advice needed | | 2021-10-02

    You could also use qubvel's segmentation models if you would like to explore semantic segmentation.

  • SonarQube

    Static code analysis for 29 languages.. Your projects are multi-language. So is SonarQube analysis. Find Bugs, Vulnerabilities, Security Hotspots, and Code Smells so you can release quality code every time. Get started analyzing your projects today for free.

  • GitHub repo segmentation_models

    Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

    Project mention: Can we use autoencoders to change an existing image instead of create one from scratch? | | 2021-12-13

    So, image segmentation (especially for satellite images) is a known problem. Search for semantic segmentation and unet (a model used for semantic segmentation). Also, if you use tensorflow there is this library which I found useful segmentation models.

  • GitHub repo pointnet2

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

    Project mention: Are the New M1 Macbooks Any Good for Deep Learning? Let’s Find Out | | 2021-02-15
  • GitHub repo Pointnet_Pointnet2_pytorch

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

    Project mention: Help me understand how to use this PointNet implementation (pytorch, point cloud classification) | | 2021-11-11

    here is a sample from one of these log files

  • GitHub repo torchio

    TorchIO: medical imaging for deep learning.

    Project mention: [P] TorchIO: a Python library for efficient loading, preprocessing, augmentation and patch-based sampling of medical images in deep learning | | 2021-01-24

    Doesn't seem that way, but its being discussed

  • GitHub repo BlenderProc

    A procedural Blender pipeline for photorealistic training image generation

    Project mention: Searching for MIT CSAIL's IKEA dataset | | 2021-12-22

    I'm trying to use BlenderProc to automatically generate training data for object recognition.

  • Scout APM

    Less time debugging, more time building. Scout APM allows you to find and fix performance issues with no hassle. Now with error monitoring and external services monitoring, Scout is a developer's best friend when it comes to application development.

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

    Project mention: 3D rcnn | | 2021-09-07

    Checkout, they have a recent 3d rcnn implementation. Their paper is a good start on the topic. Also have a look at nnDetection from the same group, it might provide some more leads. Hth

  • GitHub repo PaddleViT

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

    Project mention: Researchers From China Propose A Pale-Shaped Self-Attention (PS-Attention) And A General Vision Transformer Backbone, Called Pale Transformer | | 2022-01-09

    Code for found:

  • GitHub repo diffgram

    Complete training data platform for machine learning delivered as a single application.

    Project mention: Open Source Training Data – Diffgram | | 2021-09-07
  • GitHub repo ttach

    Image Test Time Augmentation with PyTorch!

    Project mention: Setting up Google Colab for Deep Learning | | 2021-06-09

    While Colab usually comes pre-installed with most of the basic dependencies like Tensorflow, PyTorch, scikit-learn, pandas and many more, there are chances that you have to install external packages at times. You can do that using the !pip install command. For example we can install the ttach library which is used for augmentation of images during test phase. This can be done using:

  • GitHub repo inaSpeechSegmenter

    CNN-based audio segmentation toolkit. Allows to detect speech, music and speaker gender. Has been designed for large scale gender equality studies based on speech time per gender.

    Project mention: ytmdl Web - A webapp that lets you download music by getting the audio from YouTube and metadata from various sources like Itunes, Last.FM, Gaana and others. v2 released with lots of fixes. | | 2021-02-26

    After looking for a few options, I came across inaSpeechSegmenter. It is a speech segmenter and if you pass it an audio file, it returns the time segments of noises and music.

  • GitHub repo pySBD

    🐍💯pySBD (Python Sentence Boundary Disambiguation) is a rule-based sentence boundary detection that works out-of-the-box.

    Project mention: Help with Sentence Splitting | | 2021-12-15 This will solve your case.

  • GitHub repo mmrazor

    OpenMMLab Model Compression Toolbox and Benchmark.

    Project mention: Still worrying about model compression? MMRazor may work for you. | | 2021-12-30
  • GitHub repo vocal-remover

    Vocal Remover using Deep Neural Networks

    Project mention: Bubble Phone (Hyuna, Maroon 5 feat. Wiz Khalifa) - [3:15] [OC] | | 2021-05-19

    I've wanted to mess with Bubble Pop for ages, but unfortunately, like most kpop there are no official or leaked instrumentals/acapellas/stems. Today I started experimenting with a neural network for extracting vocals and instrumentals from mastered audio (vocal-remover). Bubble Pop was the first song I tried it on, and it worked stunningly well. I got a pretty well-separated instrumental and vocal track, and I used those to make this custom mix.

  • GitHub repo BCNet

    Deep Occlusion-Aware Instance Segmentation with Overlapping BiLayers [CVPR 2021]

    Project mention: [CVPR2021] Deep Occlusion-Aware Instance Segmentation with Overlapping BiLayers | | 2021-04-10


  • GitHub repo TTNet-Real-time-Analysis-System-for-Table-Tennis-Pytorch

    Unofficial implementation of "TTNet: Real-time temporal and spatial video analysis of table tennis" (CVPR 2020)

    Project mention: Good cameras for computer vision applied to tennis | | 2021-08-30

    I'll consider using two cameras, I figured one was enough because this paper gets good results with just that and was planning to use the same/similar network to get the same/similar results but applied to a different sport.

  • GitHub repo SudachiPy

    Python version of Sudachi, a Japanese tokenizer.

    Project mention: software which turn hiragana and katakana into kanji | | 2021-08-29

    There are free tools for both of these things. I made game2text to do OCR and script matching. It includes a segmentation and normalization library Sudachi but I have not used its normalization feature for the app. I'm not sure anyone else even wants this feature but it will be pretty straightforward to add it if you're familiar with Python and vanilla Javascript.

  • GitHub repo rembg-greenscreen

    Rembg Video Virtual Green Screen Edition

    Project mention: Remove background from image or video using machine learning | | 2021-07-26
  • GitHub repo Mask-RCNN-TF2

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

    Project mention: [Video Tutorial] Run Mask R-CNN for Object Detection | | 2021-05-14
  • GitHub repo SemanticSegmentation

    A framework for training segmentation models in pytorch on labelme annotations with pretrained examples of skin, cat, and pizza topping segmentation

    Project mention: Video processing for live video using resnet, processing takes longer than each frame lasts | | 2021-02-20
  • GitHub repo comma10k-baseline

    A baseline segmentation example using the comma10k dataset (WIP)

    Project mention: EfficientNetB7 training is slow | | 2021-10-25

    Not sure if this helps but I think they have a lot of success with it here:

  • GitHub repo Orchestra

    Orchestra is a sheet music reader (optical music recognition (OMR) system) that converts sheet music to a machine-readable version.

    Project mention: [Project] Orchestra | | 2021-03-12
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 2022-01-09.

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 9,493
2 segmentation_models.pytorch 4,697
3 segmentation_models 3,620
4 pointnet2 2,167
5 Pointnet_Pointnet2_pytorch 1,305
6 torchio 1,253
7 BlenderProc 1,148
8 medicaldetectiontoolkit 1,077
9 PaddleViT 713
10 diffgram 667
11 ttach 659
12 inaSpeechSegmenter 460
13 pySBD 403
14 mmrazor 400
15 vocal-remover 347
16 BCNet 346
17 TTNet-Real-time-Analysis-System-for-Table-Tennis-Pytorch 292
18 SudachiPy 270
19 rembg-greenscreen 188
20 Mask-RCNN-TF2 132
21 SemanticSegmentation 88
22 comma10k-baseline 53
23 Orchestra 29
Find remote jobs at our new job board There are 29 new remote jobs listed recently.
Are you hiring? Post a new remote job listing for free.
OPS - Build and Run Open Source Unikernels
Quickly and easily build and deploy open source unikernels in tens of seconds. Deploy in any language to any cloud.