File size: 3,521 Bytes
faf3b5a 3468859 cd73668 106be34 faf3b5a aeb290a faf3b5a bd05d51 ea90c07 bd05d51 3468859 faf3b5a 143fc81 06c3a08 63e8ec2 06c3a08 4724f8d bd05d51 3468859 dc03340 3468859 04fe6a3 d7433ab 23660f2 d3980a6 23660f2 9450d3d bd05d51 b92c569 12f35c0 b92c569 bd05d51 3468859 faf3b5a 3468859 faf3b5a bd05d51 3468859 faf3b5a ad1ce7c bd05d51 1bdaa95 19b6500 106be34 bd05d51 3468859 faf3b5a 3468859 |
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 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
# Automatic liver segmentation in CT using deep learning
[![license](https://img.shields.io/github/license/DAVFoundation/captain-n3m0.svg?style=flat-square)](https://github.com/DAVFoundation/captain-n3m0/blob/master/LICENSE)
[![Build Actions Status](https://github.com/andreped/livermask/workflows/Build/badge.svg)](https://github.com/andreped/livermask/actions)
[![DOI](https://zenodo.org/badge/238680374.svg)](https://zenodo.org/badge/latestdoi/238680374)
<img src="figures/Segmentation_3DSlicer.PNG" width="70%" height="70%">
## Install
```
pip install git+https://github.com/andreped/livermask.git
```
As TensorFlow 2.4 only supports Python 3.6-3.8, so does livermask.
To add GPU inference support for liver vessel segmentation (which uses Chainer and CuPy), you need to install [CuPy](https://github.com/cupy/cupy). This can be easily done by adding `cupy-cudaX`, where `X` is the CUDA version you have installed, for instance `cupy-cuda110` for CUDA-11.0:
```
pip install cupy-cuda110
```
Program has been tested using Python 3.7 on Windows, macOS, and Ubuntu Linux 18.04.
## Usage:
```
livermask --input path-to-input --output path-to-output
```
| command<img width=10/> | description |
| ------------------- | ------------- |
| `--input` | the full path to the input data. Could be nifti file or directory (if directory is provided as input) |
| `--output` | the full path to the output data. Could be either output name or directory (if directory is provided as input) |
| `--cpu` | to disable the GPU (force computations on CPU only) |
| `--verbose` | to enable verbose |
| `--vessels` | to segment vessels |
### Using code directly:
If you wish to use the code directly (not as a CLI and without installing), you can run this command:
```
python -m livermask.livermask --input path-to-input --output path-to-output
```
## DICOM/NIfTI format
Pipeline assumes input is in the NIfTI format, and output a binary volume in the same format (.nii).
DICOM can be converted to NIfTI using the CLI [dcm2niix](https://github.com/rordenlab/dcm2niix), as such:
```
dcm2niix -s y -m y -d 1 "path_to_CT_folder" "output_name"
```
Note that "-d 1" assumed that "path_to_CT_folder" is the folder just before the set of DICOM scans you want to import and convert. This can be removed if you want to convert multiple ones at the same time. It is possible to set "." for "output_name", which in theory should output a file with the same name as the DICOM folder, but that doesn't seem to happen...
## Troubleshooting
You might have issues downloading the model when using VPN. If any issues are observed, try to disable VPN and try again.
If the program struggles to install, attempt to install using:
```
pip install --force-reinstall --no-deps git+https://github.com/andreped/livermask.git
```
## Acknowledgements
If you found this tool helpful in your research, please, consider citing it:
<pre>
@software{andre_pedersen_2021_5773145,
author = {André Pedersen},
title = {andreped/livermask: v1.3.0},
month = dec,
year = 2021,
publisher = {Zenodo},
version = {v1.3.0},
doi = {10.5281/zenodo.5773145},
url = {https://doi.org/10.5281/zenodo.5773145}}
</pre>
Information on how to cite can be found [here](https://zenodo.org/badge/latestdoi/238680374).
The model was trained on the LITS dataset. The dataset is openly accessible and can be downloaded from [here](https://competitions.codalab.org).
------
Made with :heart: and python
|