File size: 1,291 Bytes
760f688
 
 
 
6cb78d0
 
 
760f688
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
---
license: apache-2.0
---

# UMambaAdj: Advancing GTV Segmentation for Head and Neck Cancer in MRI-Guided RT with UMamba and nnU-Net ResEnc Planner



This repository contains the trained weights and validation results of the proposed methods for T2-weighted MRI head and neck tumor segmentation, including GTVp and GTVn segmentation for the [HNTS-MRG 2024 challenge](https://hntsmrg24.grand-challenge.org/).

Preprocessing, postprocessing and model codes can be found at [UMambaAdj Github](https://github.com/Aarhus-RadOnc-AI/UMambaAdj).

## Available Model Weights

The trained weights and validation results are stored in the following directories:
- nnUNetTrainerResenc__nnUNetResEncUNetMPlans__3d_fullres_bs4
- nnUNetTrainerUmamba__nnUNetResEncUNetMPlans__3d_fullres_bs4

These directories correspond to:
\1. nnUNetTrainerResenc: The nnU-Net Residual Encoder model with M plans.
\2. nnUNetTrainerUmamba: The UMamba model with the proposed modifications.

## How to Use

Download the trained weights from this repository.
Load the model weights into your nnU-Net environment following the standard loading instructions provided by nnU-Net.


For more details on the validation performance, refer to the [HNTS-MRG 2024 challenge](https://hntsmrg24.grand-challenge.org/) and the paper.