Semantic Segmentation on Cityscapes using Segmentation Models Pytorch
-
Updated
Jan 27, 2023 - Python
#
segmentation-models-pytorch
Here are 15 public repositories matching this topic...
A full deep learning pipeline for satellite image semantic segmentation (originally designed for detection of built structures) using Segmentation Models PyTorch.
deep-learning pytorch remote-sensing satellite-imagery semantic-segmentation segmentation-models-pytorch
-
Updated
Mar 21, 2023 - Python
This repository contains the code to train a custom DGCNN segmentation model on 3D point cloud data and carry out post-processing to filter these point clouds from the k-regular graphs produced by the model.
-
Updated
Dec 5, 2024 - Python
Implementing FCN & U-Net from scratch, then scaling with segmentation_models_pytorch — covering binary and multi-class segmentation on real-world datasets. / FCN, U-Net 논문 직접 구현부터 SMP 라이브러리 활용까지 — 실제 데이터셋으로 이진/다중 클래스 세그멘테이션을 단계적으로 학습한 기록.
python computer-vision deep-learning pytorch fcn image-segmentation unet semantic-segmentation segmentation-models-pytorch
-
Updated
Apr 21, 2026 - Python
Config-first semantic segmentation toolkit for training, evaluation, and image/video inference from reusable YAML workflows.
yaml computer-vision deep-learning transformers pytorch image-segmentation semantic-segmentation video-segmentation torchvision fastapi huggingface segmentation-models-pytorch
-
Updated
Aug 18, 2026 - Python
Deep Hierarchical Semantic Segmentation — Pixel-Wise Classification with Multi-Level Labels
open-source computer-vision deep-learning torch segmentation object-detection semantic-segmentation object-classification hierarchical-segmentation torchvision segmentation-models segmentation-models-pytorch
-
Updated
Jan 22, 2026 - Python
Multi-class brain tumor segmentation from MRI using 3D Attention U-Net
deep-learning pytorch mri medical-imaging brain-tumor brain-tumor-segmentation 3d-unet medical-imaging-with-deep-learning attention-unet segmentation-models-pytorch
-
Updated
Jul 4, 2026 - Python
Code accompanying the paper "Comprehensive evaluation of cross-cancer generalization in histopathology segmentation models across 21 tumor types." Includes annotation scripts, segmentation inference, manual scoring app, and statistical analysis.
deep-learning tcga pytorch segmentation oncology wsi computational-pathology histopathology segmentation-models-pytorch
-
Updated
May 5, 2026 - Python
This project offers an easy, flexible, modular PyTorch implementation for semantic segmentation to minimize configuration, automate training and deployment, and enable customization of models, encoders, losses and datasets through its modular design.
deep-neural-networks deep-learning image-processing pytorch image-segmentation semantic-segmentation encoder-decoder timm segmentation-models-pytorch
-
Updated
Sep 4, 2023 - Python
A General Tracking Framework for ML-based Time-Symmetric Tracking.
tracking cnn pytorch neural-networks segmentation multi-object-tracking instance-segmentation detectron2 segmentation-models-pytorch time-symmetric
-
Updated
Apr 15, 2025 - Python
基于 PyTorch 的皮肤病灶分割实验框架，支持 U-Net 系列、Kaggle GPU 训练、评估、可视化与 Gradio。 / Reproducible PyTorch framework for skin lesion segmentation with U-Net models, Kaggle GPU training, evaluation, visualization, and Gradio.
computer-vision deep-learning pytorch kaggle medical-imaging unet gradio semantic-segmentation medical-image-segmentation deeplabv3plus skin-lesion-segmentation attention-unet segmentation-models-pytorch unet-plus-plus
-
Updated
Aug 1, 2026 - Python
Custom deep learning model for 3D medical image segmentation, enhancing the UNETR architecture with lightweight attention gates on its skip connections. This proof-of-concept aimed to improve how the model combines features from different resolution levels. Experiments on MRI scans suggest this could be effective with larger training setups.
deep-learning image-processing attention-mechanism attention-is-all-you-need unet-image-segmentation segmentation-models vision-transformer segmentation-models-pytorch unetr unetr-segmentation
-
Updated
Sep 15, 2025 - Python
A hands-on PyTorch semantic segmentation lab for beginners, with runnable training, model and dataset switching, custom extensions, and bilingual guides
python education computer-vision deep-learning pytorch fcn image-segmentation beginner-friendly unet semantic-segmentation deeplabv3 torchvision segmentation-models-pytorch oxford-iiit-pet
-
Updated
Aug 22, 2026 - Python
Dense Prediction Vision Transformer (DPT) - ADE20K Reimplementation
-
Updated
Jun 26, 2025 - Python
We are Team RJAC and this repo is for the prototype phase for Edge AI Innovation Challenge 2024 hosted by DigiToad Technologies in collaboration with STMicroelectronics.
-
Updated
Oct 16, 2024 - Python
Improve this page
Add a description, image, and links to the segmentation-models-pytorch topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the segmentation-models-pytorch topic, visit your repo's landing page and select "manage topics."