diff --git a/README.md b/README.md index 3c58f08..7df325b 100644 --- a/README.md +++ b/README.md @@ -1,11 +1,15 @@ # RadFM The official code for "Towards Generalist Foundation Model for Radiology" - [ArXiv](http://arxiv.org/abs/2308.02463) + + +[ArXiv](http://arxiv.org/abs/2308.02463) + +[Website](https://chaoyi-wu.github.io/RadFM/) + +[Model checkpoint](https://huggingface.co/chaoyi-wu/RadFM) - [Website](https://chaoyi-wu.github.io/RadFM/) - ## Quick Start: Download [Model checkpoint](https://huggingface.co/chaoyi-wu/RadFM) and check `./src/test.py` for how to generate text with our model. @@ -15,14 +19,6 @@ Our pre-train code is given in ```./src/train.py```. * Check the [data_csv](https://huggingface.co/datasets/chaoyi-wu/RadFM_data_csv) (uploading) to get how different datasets are processed and down load them into `./src/Dataset/data_csv` * Modify the path as you disire, and check ```./src/train.py``` to pre-train. -## Case Study: - - - - - - - ## To-do List: - Polish the code for easier usage. - Update an easy sample for a quick start. @@ -58,6 +54,15 @@ MedKD Dataset downloading URL: | RSNA| https://www.rsna.org/education/ai-resources-and-training/ai-image-challenge/rsna-pneumonia-detection-challenge-2018| Open Access | | SIIM-ACR | https://www.kaggle.com/datasets/jesperdramsch/siim-acr-pneumothorax-segmentation-data| Open Access | +## Case Study: + +Some cases produced by our final model: + + + + + + ## Acknowledgment: We sincerely thank all the contributors who uploaded the relevant data in our dataset online. We appreciate their willingness to make these valuable cases publicly available.