/ct_mar_attention

Codes and data for paper "Rigid and Non-rigid Motion Artifact Reduction in X-ray CT using Attention Module"

Primary LanguagePython

Rigid and Non-rigid Motion Artifacts Reduction in X-ray CT using Attention Module

This repository offers the data and code introduced in the paper: "Rigid and Non-rigid Motion Artifacts Reduction in X-ray CT using Attention Module".

Dependencies

To utilize our code, you need to install followings on your system:

Usage

Download our code

To download our code, clone this repository as follow:

git clone https://github.com/youngjun-ko/ct_mar_attention
cd ct_mar_attention

Download our Dataset and Pre-trained models

Our dataset and trained models can download here

Pre-trained VGG model

Pre-trained VGG model can be downloaded from here

Training & Testing network

python train.py
python test.py

Prepare your data in following form:

  • folder name: './data'
  • file format: '.npy' (numpy array)
  • data format: 'NHWC'

Contact

E-mail: youngjun.ko@yonsei.ac.kr