https://github.com/Coradlut/Bi-JROS

The code for Bi-JROS.
https://github.com/Coradlut/Bi-JROS

Last synced: 4 months ago
JSON representation

The code for Bi-JROS.

• Host: GitHub
• URL: https://github.com/Coradlut/Bi-JROS
• Owner: Coradlut
• Created: 2024-03-27T14:52:30.000Z (10 months ago)
• Default Branch: main
• Last Pushed: 2024-04-23T13:00:06.000Z (9 months ago)
• Last Synced: 2024-07-16T00:20:47.275Z (7 months ago)
• Language: Python
• Size: 2.5 MB
• Stars: 7
• Watchers: 1
• Forks: 2
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• awesome-segmentation-and-graph-method - code

README

        
# Bi-JROS: Bi-level Learning of Task-Specific Decoders for Joint Registration and One-Shot Medical Image Segmentation

This is the source code of paper "[Bi-JROS: Bi-level Learning of Task-Specific Decoders for Joint Registration and One-Shot Medical Image Segmentation

]. 

![The proposed framework](framework.png)

## REQUIREMENTS

This code requires the following:

* Python==3.8

* PyTorch==1.12.1

* Torchvision==0.13.1

* Torchaudio==0.12.1

* Numpy==1.24.3

* Scipy==1.10.1

* Scikit-image==0.21.0

* Nibabel==5.2.0 

## USAGE

### Step 1: Getting Started

Clone the repo:

```

git clone https://github.com/Coradlut/Bi-JROS.git

```

### Step 2: Training 

```

python train.py

```

Before executing the code, it may be necessary to configure certain parameters in accordance with specific requirements.

### Step 3: Prediction

To test the performance:

```

python infer.py

```