Home

Awesome

Transformer Network for Significant Stenosis Detection in CCTA of Coronary Arteries

The implementation of our MICCAI2021 paper "Transformer Network for Significant Stenosis Detection in CCTA of Coronary Arteries".

<p align="center"> <img src="images/TR-Net.png" width="800"> </p>

Requirements

Python 3.6, PyTorch 1.6 and other common packages are listed in requirements.txt

Usage

Volume sequences can be obtained from MPR images through data_maker.py.

Cubic volumes are flattened and combined with the corresponding labels to consist of a 1D vectors, and image information sequences are obtained by concatenating these vectors.

Both training data and test data are saved as numpy arrays of shape (D, L, N^3), where D indicates the number of data on centerline-level.

The path of training data and test data can be set in config.py, for example:

train_dataset_root = './dataset/train_dataset.npy'
test_dataset_root = './dataset/test_dataset.npy'

Citation

Please consider citing the project in your publications if it helps your research. The following is a BibTeX reference. The BibTeX entry requires the url LaTeX package.

@article{ma2021transformer,
  title={Transformer Network for Significant Stenosis Detection in CCTA of Coronary Arteries},
  author={Ma, Xinghua and Luo, Gongning and Wang, Wei and Wang, Kuanquan},
  journal={arXiv preprint arXiv:2107.03035},
  year={2021}
}