Skip to content

Research project using Standford's Musculoskeletal Radiographs (MURA) dataset to detect Abnormality

Notifications You must be signed in to change notification settings

mahendrathapa/MURA

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

83 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This repo is dedicated for our attempt in research using Standford's Musculoskeletal Radiographs (MURA) dataset to detect abnormality

Set up conda environment

bash setup.sh

Activate conda environment

conda activate mura

Train Model

python main.py --env server --mode train

Inference from Model

python main.py --env server --mode predict --run_id <unique identifier> --model_checkpoint <saved model present inside out/<run_id>/checkpoints> --predict_data_dir <image dir for inference>

Back-End: Run the Flask Server

Flask serves the trained model through API, before serving, we need to update the RUN_ID and MODEL_NAME in src/constant.py

python -m src.api

Front-End

Front-End App is developed in React.

To host the front end, change to the dir:

cd src/frontend

Install React Dependency

yarn

Run Front-End

yarn start

Our Trained Model

* Download link for latest model: 
      - https://drive.google.com/drive/folders/1pnME9FFNsNJ8WJcuofn33PbExCN-XzzM?usp=sharing
* Create necessary folder structure: out/<run_id>/checkpoints
* Move the downloaded model inside: out/<run_id>/checkpoints

Citation

@ARTICLE{2017arXiv171206957R,
       author = {{Rajpurkar}, Pranav and {Irvin}, Jeremy and {Bagul}, Aarti and
         {Ding}, Daisy and {Duan}, Tony and {Mehta}, Hershel and {Yang}, Brand
        on and {Zhu}, Kaylie and {Laird}, Dillon and {Ball}, Robyn L. and
         {Langlotz}, Curtis and {Shpanskaya}, Katie and {Lungren}, Matthew P. and
         {Ng}, Andrew Y.},
        title = "{MURA: Large Dataset for Abnormality Detection in Musculoskeletal Radiographs}",
      journal = {arXiv e-prints},
     keywords = {Physics - Medical Physics, Computer Science - Artificial Intelligence},
         year = "2017",
        month = "Dec",
          eid = {arXiv:1712.06957},
        pages = {arXiv:1712.06957},
archivePrefix = {arXiv},
       eprint = {1712.06957},
 primaryClass = {physics.med-ph},
       adsurl = {https://ui.adsabs.harvard.edu/abs/2017arXiv171206957R},
      adsnote = {Provided by the SAO/NASA Astrophysics Data System}
}

About

Research project using Standford's Musculoskeletal Radiographs (MURA) dataset to detect Abnormality

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •