Projects in Awesome Lists tagged with segmentation-models

https://github.com/qubvel-org/segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

computer-vision deeplab-v3-plus deeplabv3 dpt fpn image-processing image-segmentation imagenet models pretrained-weights pspnet pytorch segformer segmentation segmentation-models semantic-segmentation transformers unet unet-pytorch unetplusplus

Last synced: 09 Sep 2025

https://github.com/qubvel/segmentation_models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

densenet efficientnet fpn image-segmentation keras keras-examples keras-models keras-tensorflow linknet mobilenet pre-trained pretrained pspnet resnet resnext segmentation segmentation-models tensorflow tensorflow-keras unet

Last synced: 13 May 2025

https://github.com/black0017/medicalzoopytorch

A pytorch-based deep learning framework for multi-modal 2D/3D medical image segmentation

3d-convolutional-network brats2018 brats2019 deep-learning densenet iseg iseg-challenge medical-image-processing medical-image-segmentation medical-imaging mrbrains18 pytorch resnet segmentation segmentation-models unet unet-image-segmentation

Last synced: 15 May 2025

https://github.com/black0017/MedicalZooPytorch

A pytorch-based deep learning framework for multi-modal 2D/3D medical image segmentation

3d-convolutional-network brats2018 brats2019 deep-learning densenet iseg iseg-challenge medical-image-processing medical-image-segmentation medical-imaging mrbrains18 pytorch resnet segmentation segmentation-models unet unet-image-segmentation

Last synced: 09 May 2025

https://github.com/guanfuchen/semseg

常用的语义分割架构结构综述以及代码复现华为媒体研究院图文Caption、OCR识别、图视文多模态理解与生成相关方向工作或实习欢迎咨询 15757172165 https://guanfuchen.github.io/media/hw_zhaopin_20220724_tiny.jpg

bisenet deep-learning deeplab drn enet erfnet fcn frrn future-semantic-segmentation gcn image-segmentation label-refinement-network lrn pytorch segmentation-models segmentation-networks segnet semantic-segmentation shufflenet

Last synced: 12 Jul 2025

https://github.com/wherobots/geotorchai

GeoTorchAI: A Framework for Training and Using Spatiotemporal Deep Learning Models at Scale

classification-model convlstm-pytorch deep-learning deep-neural-networks deepsat prediction-model raster-data satellite-classification satellite-images segmentation-models sequence-models spatial-data-analysis spatio-temporal-analysis spatio-temporal-models spatio-temporal-prediction st-resnet

Last synced: 05 Feb 2026

https://github.com/nibtehaz/MultiResUNet

MultiResUNet : Rethinking the U-Net architecture for multimodal biomedical image segmentation

medical-imaging segmentation-models

Last synced: 13 Feb 2026

https://github.com/htcr/sam_road

Segment Anything Model for large-scale, vectorized road network extraction from aerial imagery. CVPRW 2024

autonomous-driving computer-vision cvpr2024 graph graph-neural-networks graph-representation-learning mapping navigation remote-sensing scene-graph scene-graph-generation segment-anything segmentation-models transformers

Last synced: 12 Jan 2026

https://github.com/vladih/flutter_vision

A Flutter plugin for managing both Yolov5 model and Tesseract v4, accessing with TensorFlow Lite 2.x. Support object detection, segmentation and OCR on both iOS and Android.

detection-model ocr-recognition segmentation-models yolov5 yolov8

Last synced: 21 Feb 2026

https://github.com/neuropoly/totalspineseg

Robust Segmentation and Labeling of Vertebrae, Intervertebral Discs, Spinal Cord, and Spinal Canal in MRI Images Using nnU-Net and Iterative Algorithm.

3d-medical-images mri nnu-net segmentation segmentation-models spinal-canal spinal-cord spine vertebrae vertebrae-identification vertebrae-segmentation

Last synced: 12 Mar 2026

https://github.com/mbzuai-oryx/geopixel

GeoPixel: A Pixel Grounding Large Multimodal Model for Remote Sensing is specifically developed for high-resolution remote sensing image analysis, offering advanced multi-target pixel grounding capabilities.

foundation-models grounding-llms large-multimodal-models large-vision-language-models remote-sensing segmentation-models vision-language-models

Last synced: 19 Jun 2025

https://github.com/zfturbo/segmentation_models_pytorch_3d

Segmentation models for 3D data with different backbones. PyTorch.

3d 3d-models pretrained-models pytorch segmentation segmentation-models unet-segmentation

Last synced: 12 Sep 2025

https://github.com/orobix/quadra

Quadra: Effortless and reproducible deep learning workflows with configuration files.

anomaly-detection computer-vision deep-learning hydra lightning-ai pytorch pytorch-lightning segmentation-models

Last synced: 06 Feb 2026

https://github.com/fitushar/weaklysupervised-3d-classification-of-chest-ct-using-aggregated-multiresolution-segmentation-feature

This Repo contains the updated implementation of our paper "Weakly supervised 3D classification of chest CT using aggregated multi-resolution deep segmentation features", Proc. SPIE 11314, Medical Imaging 2020: Computer-Aided Diagnosis, 1131408 (16 March 2020)

3d 3d-cnn ct-imaging deep-learning densevnet3d feature-aggregation lung-disease multi-label-classification multi-label-loss multi-resolution segmentation-features segmentation-models spie-cad spie-cad-2020 tensorflow2 tfrecords weakly-supervised-classification weakly-supervised-learning weight-binary-crossentrophy weighted-loss

Last synced: 22 Apr 2025

https://github.com/prateekralhan/Segment-Anything-Streamlit

Streamlit based implementation for the The Segment Anything Model (SAM) developed by Meta AI research

facebook-research opensourceforgood python3 segment-anything segmentation-models streamlit-webapp

Last synced: 26 Sep 2025

https://github.com/koriavinash1/digitalhistopath

camelyon camelyon16 camelyon17 deep-learning digestpath digital-pathology histopathology paip segmentation-models uncertainty-quantification whole-slide-imaging

Last synced: 05 Oct 2025

https://github.com/kazuhito00/skin-clothes-hair-segmentation-using-smp

3クラス(肌、服、髪)のセマンティックセグメンテーションを実施するモデル(A model that performs semantic segmentation of 3 classes(skin, clothes, hair))

deeplab-v3-plus onnx pan pytorcch segmentation segmentation-models segmentation-models-pytorch unet-plusplus

Last synced: 15 Jul 2025

https://github.com/fitushar/WeaklySupervised-3D-Classification-of-Chest-CT-using-Aggregated-MultiResolution-Segmentation-Feature

This Repo contains the updated implementation of our paper "Weakly supervised 3D classification of chest CT using aggregated multi-resolution deep segmentation features", Proc. SPIE 11314, Medical Imaging 2020: Computer-Aided Diagnosis, 1131408 (16 March 2020)

3d 3d-cnn ct-imaging deep-learning densevnet3d feature-aggregation lung-disease multi-label-classification multi-label-loss multi-resolution segmentation-features segmentation-models spie-cad spie-cad-2020 tensorflow2 tfrecords weakly-supervised-classification weakly-supervised-learning weight-binary-crossentrophy weighted-loss

Last synced: 20 Mar 2025

https://github.com/chrise96/weighted-masks-fusion

Weighted Masks Fusion (WMF) - Ensembling for Instance Segmentation.

ensemble fusion hubmap kaggle kaggle-competition mask segmentation-models wbf yolov8

Last synced: 18 Jan 2026

https://github.com/fuzailpalnak/buildingextraction

Demo app for Building footprint extraction from satellite and aerial imagery

building-footprints building-footprints-segmentation pytorch segmentation-demo segmentation-models

Last synced: 06 Oct 2025

https://github.com/dalageo/iddroadsegmentation

Distinguishing Urban Roads from Non-Road Regions in the Indian Driving Dataset(IDD) Using Binary Deep Learning Segmentation 🛣️

cnn computer-vision deep-learning efficientnet fpn image-segmentation python pytorch segmentation-models u-net

Last synced: 04 Sep 2025

https://github.com/ashu11-a/manga-segment

Segmentation of manga in speech balloons and comics

ai bandwidth bandwidth-hero classification manga models-generation proxy proxy-server pytorch remove-background removebg segmentation segmentation-models unet unet-image-segmentation unet-segmentation yolov11 yolov8

Last synced: 07 Jun 2026

https://github.com/thedevstone/telegram-makeup-bot

Telegram bot for hair and lips makeup

dlib docker face-parsing face-recognition heroku images keras machine-learning makeup opencv segmentation-models semantic-segmentation telegram telegram-bot tensorflow2

Last synced: 13 Jul 2025

https://github.com/jabonsote/breast-segmentation-yolov8-

Este repositorio presenta una implementación de segmentación de mamas utilizando YOLOv8. La segmentación de mamas es crucial en el diagnóstico de cáncer de mama. El proyecto incluye un notebook detallado y una demostración desplegada para probar el modelo en tiempo real.

breast deep-learning segmentation segmentation-models yolo yolov8

Last synced: 21 Apr 2026

https://github.com/nata-vito/skin-cancer-segmentation

The aim of this project is to train segmentation models with png mask. This work uses U-Net model specialized in biomedical image segmentation in order to segment skin cancers.

computervision overlay segmentation-models tutorial unet-image-segmentation unet-pytorch

Last synced: 12 Jul 2025

https://github.com/ashutosh1919/context-cluster-pytorch

Ready to run PyTorch implementation of Context-Cluster: Image as Set of Points

computer-vision deep-learning image-classification machine-learning pytorch segmentation segmentation-models

Last synced: 03 Apr 2025

https://github.com/hoangtung386/brain-stroke-segmentation

Brain Stroke Segmentation in Medical Images: Advanced LCNN (Lightweight CNN) architecture for automatic detection and precise segmentation of ischemic or hemorrhagic lesions in medical scans.

attention-mechanism brain-stroke-segmentation computer-vision ct-scan-images deep-learning medical-imaging monai pytorch resnext segmentation-models symmetry-learning wandb

Last synced: 19 Apr 2026

https://github.com/ryuuzaki42/deteccao_de_linhas_de_plantio_em_plantacoes_de_cana-de-acucar_utilizando_deep_learning

Detecção de linhas de plantio em plantações de cana-de-açúcar utilizando deep learning

crop-row-detection crop-rows crop-rows-segmentation linknet pspnet python segmentation-models sugarcane tensorflow unet unet-image-segmentation

Last synced: 26 Feb 2026

https://github.com/deniztemur00/my-deep-learning-notebooks

My machine learning notebooks. Feel free to use for your purposes.

colab-notebook convolutional-neural-networks deep-learning deep-neural-networks image-segmentation isic-challenge keras-neural-networks keras-tensorflow medical-imaging python segmentation-models

Last synced: 27 Apr 2026

https://github.com/jaybfn/semantic_segmentation_project

This project was build for segmenting different body parts of a mosquito for identifying various species and combat vector-borne diseases.

computer-vision convolutional-neural-networks keras-tensorflow machine-learning neural-network python3 segmentation-models sklearn-library tensorflow2

Last synced: 13 May 2026

https://github.com/fuzailpalnak/ttaugment

Test Time Augmentation for Deep Learning Inference

augmentation computer-vision deep-learning inference segmentation segmentation-models test-time-augmentation tta

Last synced: 28 Apr 2026

https://github.com/ismailtekin05/caloriedetectingai

🍎🔍 Smart AI system that identifies food items in photos and calculates their calorie content automatically. Built with TensorFlow, YOLOv8, CUDA and computer vision for accurate nutrition tracking.

ai aimodel calorie-calculator computer-vision cuda data-analysis data-science data-segmentation data-visualization dataset dataset-generation image-processing image-recognition python segmentation-models tensorflow ultralytics yaml yolo yolov8

Last synced: 29 Apr 2026

https://github.com/radames/drawings-to-human

frontend generative-model segmentation-models sveltekit

Last synced: 19 Mar 2026

https://github.com/rathod-shubham/segmentanything

SAM is a deep learning model (transformer based). When we give an image as input to the Segment Anything Model, it first passes through an image encoder and produces a one-time embedding for the entire image. The downsampling happens using 2D convolutional layers. Then the model concatenates it with the image embedding to get the final vector.

deep-learning encoder-decoder instance-segmentation neural-network object-detection sam segment-anything segment-anything-model segmentation segmentation-models semantic-segmentation transformer

Last synced: 15 Mar 2025

https://github.com/ahestevenz/flood-detection-segmentation

Flood detector using effientnet as semantic segmentation model

computer-vision effientnet flood-detection floods machine-learning segmentation-models

Last synced: 01 Apr 2026

https://github.com/phuocphat1005/transunet_segmentation_in_medical_image

This is final project about TransUNet, an approach for Segmentation in Medical Image.

artificial-intelligence artificial-neural-networks computer-vision medical-imaging segmentation-models

Last synced: 13 Apr 2025

https://github.com/piero24/lossfuncition-4sn

This research will show an innovative method useful in the segmentation of polyps during the screening phases of colonoscopies. To do this we have adopted a new approach which consists in merging the hybrid semantic network (HSNet) architecture model with the Reagion-wise(RW) as a loss function for the backpropagation process.

dataset deep-learning deep-neural-networks loss loss-functions matplotlib neural-network numpy opencv polyp-detection polyp-segmentation python python3 pytorch segmentation segmentation-models torchvision transformers unipd unipd-dei

Last synced: 06 May 2026

https://github.com/stupidcucumber/deeplabv3-celebhq

Trained Torch version of the DeepLab on the CelebHQ.

deep-neural-networks deeplabv3 face-segmentation python3 segmentation-models torch

Last synced: 15 May 2026

https://github.com/rmd926/nycu-dlp

NYCU Deep Learning 2024 Summer Semester

computer-vision deep-learning diffusion-models nycu-course pytorch segmentation-models variational-autoencoder

Last synced: 18 Apr 2026

https://github.com/sovit-123/torch-segment

A deep learning image segmentation library and API on top of PyTorch.

convolutional-neural-networks deep-learning deep-neural-networks fcn imagenet imagesegmentation neural-network pretrained-models pytorch segmentation segmentation-masks segmentation-models unet unet-image-segmentation unet-pytorch

Last synced: 25 Apr 2026

https://github.com/kons-5/ist-aaut

This repository contains lab materials for the IST-AAut course.

arx-model cnn-classification jupyter-notebooks machine-learning machine-learning-algorithms optuna outlier-removal pytorch pytorch-lightning regression-models segmentation-models sklearn statsmodels

Last synced: 06 May 2026

https://github.com/tusharpandey003/coronarysegmentationandstenosis

Project is under development

angiograms cardiovascular-diseases deeplearning multitask-learning segmentation-models stenosis-detection

Last synced: 18 Aug 2025

https://github.com/msikorski93/retinal-vessel-segmentation-using-deep-learning

Retinal vessel segmentation is the task of segmenting vessels in retina imagery. This binary task was performed with a U-Net network.

computer-vision efficientnet healthcare imbalanced-data keras retinal-images retinal-vessel-segmentation segmentation-models tensorflow u-net

Last synced: 14 May 2026

https://github.com/rohanchawla972/market-segmentation-customer-profile-modelling

Customer segmentation and profiling repository applying clustering, regression, and statistical modeling to build buyer personas and data-driven marketing strategies.

clustering-analysis econometrics machine-learning predictive-modeling r segmentation-models spss

Last synced: 04 Oct 2025

https://github.com/dineshthapax/road-segmentation-for-autonomous-car-using-mesnet

A deep learning-based road segmentation model for autonomous vehicles, MESNet integrates ResNet-50, VGG-16, and PSPNet to achieve high accuracy and precision in diverse environments. Trained on the KITTI dataset, it handles real-time road segmentation tasks with robust performance metrics.

autonomous-car autonomous-driving autonomous-vehicles mesnet roadsegmentation segmentation segmentation-model segmentation-models