Top 23 Python image-segmentation Projects

• albumentations

  29 13,451 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: Recapping the AI, Machine Learning and Data Science Meetup — May 8, 2024 | dev.to | 2024-05-09

In this brief walkthrough, I will illustrate how to leverage open-source FiftyOne and Anomalib to build deployment-ready anomaly detection models. First, we will load and visualize the MVTec AD dataset in the FiftyOne App. Next, we will use Albumentations to test out augmentation techniques. We will then train an anomaly detection model with Anomalib and evaluate the model with FiftyOne.

• segmentation_models.pytorch

  14 8,862 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.

  

• PaddleSeg

  17 8,284 7.4 Python

  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.

  

• U-2-Net

  30 8,134 3.1 Python

  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: I used the ChatGPT API to create a proof-of-concept AI driven video game. Using generative AI for the images and dialogue and GPT-3.5 for narrative and game control. More info in comments. | /r/ChatGPT | 2023-06-17

I use a finetuned custom Stable Diffusion model in combination with a style embedding for the characters for image generation and U²-Net for background removal.

• mmsegmentation

  7 7,436 8.2 Python

  OpenMMLab Semantic Segmentation Toolbox and Benchmark.

  

• segmentation_models

  8 4,620 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

• catalyst

  1 3,229 0.0 Python

  Accelerated deep learning R&D (by catalyst-team)

  

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

• SaaSHub

  www.saashub.com featured

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

  

• ailia-models

  4 1,840 9.8 Python

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

  

• pytorch-toolbelt

  1 1,486 8.1 Python

  PyTorch extensions for fast R&D prototyping and Kaggle farming

• PixelLib

  3 1,016 0.0 Python

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

• cleanvision

  4 925 7.3 Python

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

  

• caer

  8 749 0.0 Python

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

• cloth-segmentation

  2 480 0.0 Python

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

• signature_extractor

  1 426 0.0 Python

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

• APE

  2 424 8.1 Python

  [CVPR 2024] Aligning and Prompting Everything All at Once for Universal Visual Perception (by shenyunhang)

Project mention: Image-to-Caption Generator | /r/computervision | 2023-12-07

https://github.com/shenyunhang/APE (super new, idk usability on this one yet)

• ISP-Guide

  1 283 5.0 Python

  Image Signal Processing (ISP) Guide. Learn all about the process of converting an image/video into digital form by performing tasks like noise reduction, filtering, auto exposure, autofocus, HDR correction, and image sharpening with a Specialized type of media processor.

• detectree

  1 206 7.9 Python

  Tree detection from aerial imagery in Python

• smriprep

  1 124 9.3 Python

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

  

• SLiMe

  1 106 6.6 Python

  1-shot image segmentation using Stable Diffusion (by aliasgharkhani)

Project mention: SLiMe: One-shot image segmentation using Stable Diffusion | news.ycombinator.com | 2023-10-18

• uformer-pytorch

  1 91 0.0 Python

  Implementation of Uformer, Attention-based Unet, in Pytorch

• FedCV

  2 62 2.7 Python

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

  

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

  1 43 0.0 Python

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

• Military-Vehicles-Image-Recognition

  1 43 0.0 Python

  Fine-Tuning YOLOv5 to detect Military Vehicles in Aerial ARMA 3 Imagery

Project mention: Jarvis: A Voice Virtual Assistant in Python (OpenAI, ElevenLabs, Deepgram) | news.ycombinator.com | 2023-12-18

https://github.com/AlexandreSajus/Military-Vehicles-Image-Re...

There's a French defense company, Preligens, that actually does this currently

• SaaSHub

  www.saashub.com featured

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

  

Python image-segmentation related posts

• Instance segmentation of small objects in grainy drone imagery

  8 projects | /r/computervision | 9 Dec 2023

• Image-to-Caption Generator

  3 projects | /r/computervision | 7 Dec 2023

• I used the ChatGPT API to create a proof-of-concept AI driven video game. Using generative AI for the images and dialogue and GPT-3.5 for narrative and game control. More info in comments.

  1 project | /r/ChatGPT | 17 Jun 2023

• OMPR V0.6.8 update

  1 project | /r/OMPR_PCVR | 20 Feb 2023

• [Medical Segmentation] The all-in-one 3D medical image segmentation toolkit. From data annotation to model deployment, you are welcome to try it all！

  1 project | /r/ArtificialInteligence | 19 Dec 2022

• [N][R] SOTA Real-Time Semantic Segmentation Model Implemented by Python

  1 project | /r/Python | 12 Dec 2022

• [P] New Features of Image Segmentation Project PaddleSeg (6k stars)

  1 project | /r/artificial | 1 Dec 2022
• A note from our sponsor - InfluxDB
  www.influxdata.com | 10 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

Do not miss the trending Python projects with our weekly report!