Spaces:
Running
Running
File size: 1,837 Bytes
09d466f 470e720 5052a62 b987d49 41491f8 73198ea 41491f8 5052a62 41491f8 b987d49 41491f8 5052a62 41491f8 5052a62 41491f8 |
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 32 33 34 35 36 37 38 |
# Automatic lung tumor segmentation in CT
This is the official repository for the paper [_"Teacher-Student Architecture for Mixed Supervised Lung Tumor Segmentation"_](https://arxiv.org/abs/2112.11541), **accepted for publication** in PLOS ONE.
A pretrained model is made available in a command line tool and can be used as you please. However, the current model is not intended for clinical use. The model is the result of a proof-of-concept study. An improved model will be made available in the future, when more training data is made available.
![sample of masked output](https://github.com/VemundFredriksen/LungTumorMask/releases/download/0.0.1/sample_images.png "Sample output of two different tumors")
![sample of 3d render](https://github.com/VemundFredriksen/LungTumorMask/releases/download/0.0.1/sample_renders.png "3D render of two masked outputs")
## Dependencies
In addition to the python packages specified in requirements.txt, [PyTorch](https://pytorch.org/get-started/locally/) and [lungmask](https://github.com/JoHof/lungmask) must be installed.
## Installation
```
pip install git+https://github.com/VemundFredriksen/LungTumorMask
```
## Usage
After install, the software can be used as a command line tool. Simply specify the input and output filenames to run:
```
# Format
lungtumormask input_file output_file
# Example
lungtumormask patient_01.nii.gz mask_01.nii.gz
```
## Acknowledgements
If you found this repository useful in your study, please, cite the following paper:
```
@misc{fredriksen2021teacherstudent,
title={Teacher-Student Architecture for Mixed Supervised Lung Tumor Segmentation},
author={Vemund Fredriksen and Svein Ole M. Svele and André Pedersen and Thomas Langø and Gabriel Kiss and Frank Lindseth},
year={2021},
eprint={2112.11541},
archivePrefix={arXiv},
primaryClass={eess.IV}}
```
|