Skip to content
#

segmentation-models-pytorch

Here are 22 public repositories matching this topic...

Land-Cover-Semantic-Segmentation-PyTorch

🛣 Building an end-to-end Promptable Semantic Segmentation (Computer Vision) project from training to inferencing a model on LandCover.ai data (Satellite Imagery).

  • Updated May 15, 2025
  • Jupyter Notebook

Segmentation of cancerous tumors using Mamba. Code, resources, and paper provided. We manage to make a small (42k param) model that can segment pretty well.

  • Updated May 16, 2024
  • Jupyter Notebook

This repository contains code and results for a semantic segmentation competition using a small autonomous driving dataset. The project explores various data augmentation, network architectures, and training strategies to improve segmentation accuracy and efficiency.

  • Updated Jun 25, 2025
  • Jupyter Notebook

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 라이브러리 활용까지 — 실제 데이터셋으로 이진/다중 클래스 세그멘테이션을 단계적으로 학습한 기록.

  • Updated Apr 21, 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.

  • Updated Sep 4, 2023
  • 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.

  • Updated Sep 15, 2025
  • 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.

Curate this topic

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."

Learn more