Skip to content
#

segmentation-models-pytorch

Here are 8 public repositories matching this topic...

Language: Python
Filter by language

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