1 Star 0 Fork 0

麦穗 / sarcopenia-ai

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README
BSD-3-Clause

Sarcopenia AI

Code for papers:

Kanavati, F., Islam, S., Aboagye, E. O., & Rockall, A. (2018). Automatic L3 slice detection in 3D CT images using fully-convolutional networks. arXiv preprint arXiv:1811.09244.

Kanavati, F., Islam, S., Arain, Z., Aboagye, E. O., & Rockall, A. (2020). Fully-automated deep learning slice-based muscle estimation from CT images for sarcopenia assessment. arXiv preprint arXiv:2006.06432.

Models

Trained models for slice detection and slice segmentation are provided in models/

Dev

conda create -y --name sarcopenia-ai python=3.6.2

Install

pip install -e .

Docker

Build docker image

docker build -t sarcopeniaai -f ./Dockerfile .

Slice detection trainer

Download the training data from here to your data folder.

docker run --rm -it  -v <your_data_folder>:/data -v $(pwd)/configs:/configs sarcopeniaai python -m sarcopenia_ai.apps.slice_detection.trainer --config /configs/slice_detection.cfg

Training output preview on validation images

Segmentation trainer

Labelled segmentation data is not provided. Once you get your own data, you can train a segmentation model with

docker run --rm -it -v <your_data_folder>:/data -v $(pwd)/configs:/configs sarcopeniaai python -m sarcopenia_ai.apps.segmentation.trainer --config /configs/segmentation.cfg

Run as API server

docker run --rm -it -p 5000:5000 sarcopeniaai python -m sarcopenia_ai.apps.server.run_local_server

Then head to http://localhost:5000 for web UI

You can also get results from command line. Example:

curl -X POST -F image=@data/volume.nii.gz http://localhost:5000/predict

Expected result

{
   "prediction":{
      "id":"64667bf3482d4ee5a0e8af6c67b2fa0d",
      "muscle_area":"520.15",
      "muscle_attenuation":"56.00 HU",
      "slice_prob":"69.74%",
      "slice_z":90,
      "str":"Slice detected at position 90 of 198 with 69.74% confidence "
   },
   "success":true
}

L3 annotated dataset

The dataset was collected from multiple sources:

  1. 3 sets were obtained from the Cancer Imaging Archive (TCIA):

  2. a liver tumour dataset was obtained from the LiTS segmentation challenge.

The dataset is available for download in MIPs format from here.

The subset of transitional vertabrae cases can be downloaded from here.

@article{kanavati2018automatic,
  title={Automatic L3 slice detection in 3D CT images using fully-convolutional networks},
  author={Kanavati, Fahdi and Islam, Shah and Aboagye, Eric O and Rockall, Andrea},
  journal={arXiv preprint arXiv:1811.09244},
  year={2018}
}


@article{kanavati2020fullyautomated,
    title={Fully-automated deep learning slice-based muscle estimation from CT images for sarcopenia assessment},
    author={Fahdi Kanavati and Shah Islam and Zohaib Arain and Eric O. Aboagye and Andrea Rockall},
    year={2020},
    eprint={2006.06432},
    archivePrefix={arXiv},
    primaryClass={eess.IV}
}
BSD License Copyright (c) 2018, Fahdi Kanavati All rights reserved. Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met: * Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution. * Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

简介

暂无描述 展开 收起
BSD-3-Clause
取消

发行版

暂无发行版

贡献者

全部

近期动态

加载更多
不能加载更多了
1
https://gitee.com/tutu96177/sarcopenia-ai.git
git@gitee.com:tutu96177/sarcopenia-ai.git
tutu96177
sarcopenia-ai
sarcopenia-ai
master

搜索帮助