This is the official Pytorch implementation of "Segmenting two-dimensional structures with strided tensor networks", Raghavendra Selvan et al. 2021
- Reproduce results on Lung CXR dataset reported in the paper
- v1.0
- Basic Pytorch dependency
- Tested on Pytorch 1.6, Python 3.6
- The preprocessed lungCXR data is provided as an archive here. Unzip and point that as the data_dir.
- Train the model from scratch: python train.py --data data/lungCXR
- Kindly cite our publication if you use any part of the code
@inproceedings{raghav2021StridedTenet,
title={Segmenting two-dimensional structures with strided tensor networks},
author={Raghavendra Selvan and Erik B. Dam and Jens Petersen},
booktitle={27th international conference on Information Processing in Medical Imaging (IPMI)},
month={Feb},
note={arXiv preprint arXiv:2102.06900},
year={2021}}