Skip to content
#

segmentation-models-pytorch

Here are 17 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

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