Git Product home page Git Product logo

chestrayxnet's Introduction

Diagnosis of pulmonary disease based on NIH ChestX-ray14

This project is a chest X ray images muti-label classification algorithm with high performance which have potential to help radiologist diagnosis pulmonary disease better.
The baseline of this algorithm is CheXNet, I learn lots of things the from the paper which help me train my DenseNet121 well.
All the model trained on NIH ChestX-ray14 dataset, you can download the data here

Quick Start

  • Download ChestX-ray14 dataset here, unzip all 14 xxx.tar.gz file to a single floder, run tar -xvzf *ZIPFILENAME* in terminal, ZIPFILENAME is the .tar.gz file you want to unzip.
  • Modify the image path in shell/write_all.sh, image path fellowed by --dataset_dir tag.
  • Change your work directory to ChestRayXNet.
  • Run shell/write_all.sh to convert all 112,120 chest X ray images into .tfrecord format.
  • Run shell/train_densenet_121.sh to start training a network named CheXNet. I recommend start by running train_densenet_121.sh as this script may have more additional features and less bug.
  • Futher, if you want to try different networks, modify the network type in shell/train_generate.sh fellowed by --model_type tag and log directory. If you get OOM error during training, modify the batch size fellowed by --batch_size tag.
  • To Visualize training procedure, run tensorbord --logdir='*LOG_DIR*' --port=6006 in terminal, LOG_DIR is the dir set after --log_dir tag in the shell script you running.
  • To evaluate the model performance, run shell/eval_muti.sh, if your train different network other than DenseNet121, modify the network type and log dir in eval_muti.sh file.

Acknowledgements

1.The baseline model of this project is CheXNet published by Andrew Ng et al.
2.All network writen by slim are provided by tensorflow contributers and available on github, and related pretrained model can be find here
3.A baseline project fine-tuning from ImageNet and trained on Oxford-flower dataset, I'm searching for the github link of that project.
4.I add weight to loss function fellowing this paper published by Buda et al.
5.Two earlier research on NIH ChestX-ray14 also provide many insight to me, they can be find here and here.

Last but not least

This README file write in a hurry and I belive I must make some mistake, please let me know if thier is any word that made you confused.
I update this README.md file 3 days after first commit but I think it's still not good and clear enough, maybe next version would be better.

This project is writen by Ruiqi Sun(孙瑞琦

chestrayxnet's People

Contributors

chasewinds avatar

Stargazers

 avatar  avatar  avatar Dingzhu Lurong avatar  avatar unkown avatar  avatar

Watchers

James Cloos avatar  avatar paper2code - bot avatar

Forkers

noicfank stevenji

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.