segmentation-models

There are 11 repositories under segmentation-models topic.

qubvel-org / segmentation_models.pytorch
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
computer-vision deeplab-v3-plus deeplabv3 fpn image-processing image-segmentation imagenet models pretrained-models pretrained-weights pspnet pytorch segformer segmentation segmentation-models semantic-segmentation transformers unet unet-pytorch unetplusplus
Language:Python 10177
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
Language:Python 4818
MedicalZooPytorch
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
Language:Python 1791
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
Language:Python 777
GeoTorchAI
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-images segmentation-models sequence-models spatial-data-analysis spatio-temporal-analysis spatio-temporal-models spatio-temporal-prediction st-resnet satellite-classification
Language:Jupyter Notebook 495
nibtehaz / MultiResUNet
MultiResUNet : Rethinking the U-Net architecture for multimodal biomedical image segmentation
medical-imaging segmentation-models
Language:Jupyter Notebook 450
CPM
VinAIResearch / CPM
💄 Lipstick ain't enough: Beyond Color-Matching for In-the-Wild Makeup Transfer (CVPR 2021)
makeup-transfer gan cyclegan makeup color cvpr2021 segmentation-models color-matching face-manipulation computer-vision cvpr color-transfer pattern-detection
Language:Python 387
NITR098 / Awesome-U-Net
Official repo for Medical Image Segmentation Review: The Success of U-Net
medical-image-processing medical-image-segmentation segmentation-models transformer u-net unet-image-segmentation unet-pytorch
Language:Jupyter Notebook 313
ecsplendid / rembg-greenscreen
Rembg Video Virtual Green Screen Edition
video greenscreen segmentation segmentation-models deeplearning
Language:Python 256
junqiangchen / PytorchDeepLearing
Meidcal Image Segmentation Pytorch Version
classification-model flask flask-application pytorch-implementation segmentation-models unet-image-segmentation vnet3d
Language:Python 252
DebeshJha / 2020-CBMS-DoubleU-Net
Official implementation of DoubleU-Net for Semantic Image Segmentation in TensorFlow & Pytorch (Nominated for Best Paper Award (IEEE CBMS))
semantic-segmentation cnn-keras u-net doubleunet unet-image-segmentation tensoflow doubleu-net segmentation-models pytorch-implementation convolutional-neural-networks deep-learning image-segmentation medical-imaging unet
Language:Python 239
htcr / sam_road
Segment Anything Model for large-scale, vectorized road network extraction from aerial imagery. CVPRW 2024
autonomous-driving computer-vision graph graph-neural-networks mapping navigation remote-sensing segment-anything segmentation-models transformers cvpr2024 graph-representation-learning scene-graph scene-graph-generation
Language:Python 173
TensorFlow-Advanced-Segmentation-Models
JanMarcelKezmann / TensorFlow-Advanced-Segmentation-Models
A Python Library for High-Level Semantic Segmentation Models based on TensorFlow and Keras with pretrained backbones.
segmentation-models tensorflow computer-vision semantic-segmentation segmentation-model unet segmentation pspnet fpn fcn deeplab danet ocnet cfnet acfnet ocrnet hrnet image-segmentation tensorflow-macos
Language:Python 157
NEU-DET-Steel-Surface-Defect-Detection
siddhartamukherjee / NEU-DET-Steel-Surface-Defect-Detection
This project is about detecting defects on steel surface using Unet. The dataset used for this project is the NEU-DET database.
neu-det defect-detection defect-prediction segmentation-models unet-image-segmentation unet-pytorch resnet resnet-18 pytorch deep-learning deep-neural-networks
Language:Jupyter Notebook 105
rezazad68 / TMUnet
Contextual Attention Network: Transformer Meets U-Net
deep-learning semantic-segmentation pytorch transformers unet skin-cancer pathology-image unet-image-segmentation segmentation-models
Language:Jupyter Notebook 95
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
Language:Java 79
faizan1234567 / Brain-Tumors-Segmentation
Multimodal Brain mpMRI segmentation on BraTS 2023 and BraTS 2021 datasets.
deep-learning brain-tumor-segmentation brats2020 medical-imaging 3d-segmentation computer-vision machine-learning medical-image-processing brats17 brats18 brats16 segmentation brats23 btcv organs segmentation-models synapse
Language:Python 73
Kaiseem / SLAug
[AAAI 2023] Official PyTorch implementation of the paper "SLAug: Rethinking Data Augmentation for Single-source Domain Generalization in Medical Image Segmentation"
aaai2023 domain-generalization medical-image-segmentation single-source-domain-generalization domain domain-adaptation segmentation-models
Language:Jupyter Notebook 72
luisCartoGeo / GeoAI_Plugin
Plugin para ejecutar el modelo de META Segment Anything en QGIS
python3 qgis3-plugin sam segmentation-models torch
Language:Python 71
DoctorKey / C-3PO
[PR] How to Reduce Change Detection to Semantic Segmentation
change-detection pytorch-implementation scene-change-detection segmentation-models
Language:Python 70
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
Language:Python 63
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
Language:Python 60
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
Language:Python 48
jcwang123 / xboundformer
[TMI 2023] XBound-Former: Toward Cross-scale Boundary Modeling in Transformers
medical-image-processing segmentation-models transformers-models
Language:Python 47
avakanski / Attention-Enriched-DL-Model-for-Breast-Tumor-Segmentation
Salient attention U-Net model for tumor segmentation in breast ultrasound images, based on visual saliency maps
deep-learning segmentation-models medical-imaging
Language:Jupyter Notebook 42
neuropoly / totalspineseg
Robust Segmentation and Labeling of Vertebrae, Intervertebral Discs, Spinal Cord, and Spinal Canal in MRI Images Using nnU-Net and Iterative Algorithm.
mri segmentation 3d-medical-images nnu-net segmentation-models spinal-cord spine vertebrae vertebrae-identification vertebrae-segmentation spinal-canal
Language:Python 39
Kazuhito00 / Skin-Clothes-Hair-Segmentation-using-SMP
3クラス(肌、服、髪)のセマンティックセグメンテーションを実施するモデル(A model that performs semantic segmentation of 3 classes(skin, clothes, hair))
pytorcch segmentation-models segmentation segmentation-models-pytorch unet-plusplus deeplab-v3-plus pan onnx
Language:Jupyter Notebook 36
gopinath-balu / computer_vision
Computer vision models
image-classification image-recognition object-detection segmentation-models
Language:Python 34
hamdiboukamcha / SPEED-SAM-C-TENSORRT
SPEED SAM C++ TENSORRT: A high-performance C++ implementation using TensorRT and CUDA
segment-anything segment-anything-model segmentation segmentation-models
Language:C++ 28
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
Language:Python 26
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)
ct-imaging deep-learning 3d 3d-cnn segmentation-models segmentation-features tensorflow2 weakly-supervised-learning weakly-supervised-classification densevnet3d lung-disease feature-aggregation multi-resolution tfrecords multi-label-classification multi-label-loss weighted-loss weight-binary-crossentrophy spie-cad spie-cad-2020
Language:Python 24
koriavinash1 / DigitalHistoPath
histopathology digital-pathology whole-slide-imaging deep-learning uncertainty-quantification digestpath paip camelyon16 camelyon camelyon17 segmentation-models
Language:Jupyter Notebook 24
saadwazir / HistoSeg
HistoSeg is an Encoder-Decoder DCNN which utilizes the novel Quick Attention Modules and Multi Loss function to generate segmentation masks from histopathological images with greater accuracy. This repo contains the code to Test and Train the HistoSeg
monuseg monuseg-challenge histological-images histology-images segmentation semantic-segmentation attention-mechanism self-attention nuclie-segmentation dice-loss focal-loss medical-image-segmentation image-segmentation glas segmentation-models unet deeplab-v3-plus unet-segmentation histological-image-segmentation
Language:Python 23
rezazad68 / transnorm
transnorm
pytorch deep-learning medical-image-processing medical-image-segmentation segmentation-models semantic-segmentation skin-cancer supervised-learning synapse transformer unet unet-image-segmentation
Language:Jupyter Notebook 21
AbhishekRS4 / HTSM_Oil_Spill_Segmentation
HTSM Masterwork
oil-spill segmentation-based-detection segmentation-models semantic-segmentation deep-learning encoder-decoder-model supervised-learning computer-vision docker-deployment streamlit-application
Language:Jupyter Notebook 18
karasawatakumi / monodepth-dev
Monocular depth estimation by segmentation models for NYU-depth v2 dataset. Simple PyTorch Lightning implementation.
monocular-depth-estimation segmentation-models nyu2 docker docker-compose pytorch pytorch-lightning monodepth omegaconf
Language:Python 18

segmentation-models

qubvel-org / segmentation_models.pytorch

qubvel / segmentation_models

black0017 / MedicalZooPytorch

guanfuchen / semseg

wherobots / GeoTorchAI

nibtehaz / MultiResUNet

VinAIResearch / CPM

NITR098 / Awesome-U-Net

ecsplendid / rembg-greenscreen

junqiangchen / PytorchDeepLearing

DebeshJha / 2020-CBMS-DoubleU-Net

htcr / sam_road

JanMarcelKezmann / TensorFlow-Advanced-Segmentation-Models

siddhartamukherjee / NEU-DET-Steel-Surface-Defect-Detection

rezazad68 / TMUnet

vladiH / flutter_vision

faizan1234567 / Brain-Tumors-Segmentation

Kaiseem / SLAug

luisCartoGeo / GeoAI_Plugin

DoctorKey / C-3PO

mbzuai-oryx / GeoPixel

ZFTurbo / segmentation_models_pytorch_3d

orobix / quadra

jcwang123 / xboundformer

avakanski / Attention-Enriched-DL-Model-for-Breast-Tumor-Segmentation

neuropoly / totalspineseg

Kazuhito00 / Skin-Clothes-Hair-Segmentation-using-SMP

gopinath-balu / computer_vision

hamdiboukamcha / SPEED-SAM-C-TENSORRT

prateekralhan / Segment-Anything-Streamlit

fitushar / WeaklySupervised-3D-Classification-of-Chest-CT-using-Aggregated-MultiResolution-Segmentation-Feature

koriavinash1 / DigitalHistoPath

saadwazir / HistoSeg

rezazad68 / transnorm

AbhishekRS4 / HTSM_Oil_Spill_Segmentation

karasawatakumi / monodepth-dev