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 9711
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 4764
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 1738
guanfuchen / semseg
常用的语义分割架构结构综述以及代码复现华为媒体研究院图文Caption、OCR识别、图视文多模态理解与生成相关方向工作或实习欢迎咨询 15757172165 https://guanfuchen.github.io/media/hw_zhaopin_20220724_tiny.jpg
semantic-segmentation deep-learning pytorch fcn segnet erfnet enet drn lrn label-refinement-network image-segmentation segmentation-networks segmentation-models deeplab bisenet frrn shufflenet future-semantic-segmentation gcn
Language:Python 770
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-classification satellite-images segmentation-models sequence-models spatial-data-analysis spatio-temporal-analysis spatio-temporal-models spatio-temporal-prediction st-resnet
Language:Jupyter Notebook 484
nibtehaz / MultiResUNet
MultiResUNet : Rethinking the U-Net architecture for multimodal biomedical image segmentation
medical-imaging segmentation-models
Language:Jupyter Notebook 441
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 374
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 280
ecsplendid / rembg-greenscreen
Rembg Video Virtual Green Screen Edition
deeplearning greenscreen segmentation segmentation-models video
Language:Python 254
DebeshJha / 2020-CBMS-DoubleU-Net
Official implementation of DoubleU-Net for Semantic Image Segmentation in TensorFlow & Pytorch (Nominated for Best Paper Award (IEEE CBMS))
cnn-keras convolutional-neural-networks deep-learning doubleu-net doubleunet image-segmentation medical-imaging pytorch-implementation segmentation-models semantic-segmentation tensoflow u-net unet unet-image-segmentation
Language:Python 239
junqiangchen / PytorchDeepLearing
Meidcal Image Segmentation Pytorch Version
classification-model pytorch-implementation segmentation-models unet-image-segmentation vnet3d flask flask-application
Language:Python 228
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 152
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 148
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 93
rezazad68 / TMUnet
Contextual Attention Network: Transformer Meets U-Net
deep-learning pathology-image pytorch segmentation-models semantic-segmentation skin-cancer transformers unet unet-image-segmentation
Language:Jupyter Notebook 92
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 74
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 68
luisCartoGeo / GeoAI_Plugin
Plugin para ejecutar el modelo de META Segment Anything en QGIS
python3 qgis3-plugin sam segmentation-models torch
Language:Python 67
DoctorKey / C-3PO
[PR] How to Reduce Change Detection to Semantic Segmentation
change-detection pytorch-implementation scene-change-detection segmentation-models
Language:Python 63
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 47
jcwang123 / xboundformer
[TMI 2023] XBound-Former: Toward Cross-scale Boundary Modeling in Transformers
medical-image-processing segmentation-models transformers-models
Language:Python 42
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 medical-imaging segmentation-models
Language:Jupyter Notebook 40
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 36
gopinath-balu / computer_vision
Computer vision models
image-classification image-recognition object-detection segmentation-models
Language:Python 34
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 34
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
koriavinash1 / DigitalHistoPath
histopathology digital-pathology whole-slide-imaging deep-learning uncertainty-quantification digestpath paip camelyon16 camelyon camelyon17 segmentation-models
Language:Jupyter Notebook 23
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 22
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 21
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 20
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
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 18
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 16
AxDante / SAAMI
Automatic segment and generate masks for any 3D medical images using SAM model without prompt
3d-segmentation medical-image-segmentation medical-imaging segment-anything segment-anything-model segmentation-models medical sam saami mri mri-segmentation
Language:Python 14
rachitsaluja / BraTS-2023-Metrics
Official BraTS 2023 Segmentation Performance Metrics
brats brats-challenge deep-learning performance-metrics segmentation segmentation-models
Language:Python 13
naamiinepal / vlsm-adapter
VLSM-Adapter: Finetuning Vision-Language Segmentation Efficiently with Lightweight Blocks [MICCAI 2024]
adapter segmentation-models transfer-learning vision-language-model
Language:Python 10

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

DebeshJha / 2020-CBMS-DoubleU-Net

junqiangchen / PytorchDeepLearing

JanMarcelKezmann / TensorFlow-Advanced-Segmentation-Models

htcr / sam_road

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

rezazad68 / TMUnet

vladiH / flutter_vision

Kaiseem / SLAug

luisCartoGeo / GeoAI_Plugin

DoctorKey / C-3PO

orobix / quadra

jcwang123 / xboundformer

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

ZFTurbo / segmentation_models_pytorch_3d

gopinath-balu / computer_vision

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

prateekralhan / Segment-Anything-Streamlit

koriavinash1 / DigitalHistoPath

saadwazir / HistoSeg

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

rezazad68 / transnorm

karasawatakumi / monodepth-dev

neuropoly / totalspineseg

AbhishekRS4 / HTSM_Oil_Spill_Segmentation

AxDante / SAAMI

rachitsaluja / BraTS-2023-Metrics

naamiinepal / vlsm-adapter