YenJung commited on
Commit
21cff44
·
1 Parent(s): 18f919c

Delete README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -24
README.md DELETED
@@ -1,24 +0,0 @@
1
- # HeartNet
2
-
3
- A joint project by [oapostrophe](https://github.com/oapostrophe), [gkenderova](https://github.com/gkenderova), [soksamnanglim](https://github.com/soksamnanglim), [syaa2018](https://github.com/syaa2018)
4
-
5
- For a high-level overview of this project, check out this [blog post](https://oapostrophe.github.io/heartnet/) and [90-second demo](https://www.youtube.com/watch?v=EqAU-FRu6C4). For a full presentation and more detailed writeup on our methodology, check out the report on our [project website](https://oapostrophe.github.io/HeartNet/).
6
-
7
- The trained model can be demoed by downloading `app.py` and `demo_model.pkl`, installing [streamlit](https://anaconda.org/conda-forge/streamlit) and [fastai](https://pypi.org/project/fastai/), then running:
8
- ```shell
9
- streamlit run app.py
10
- ```
11
- You can then visit the provided url in your browser; for convenience, sample generated MI and Normal EKG images are provided in the `/test files` directory.
12
-
13
- To use any of the other files, you'll have to download the [PTB-XL](https://physionet.org/content/ptb-xl/1.0.1/) dataset.
14
-
15
- The important files are the following:
16
- - `app.py` StreamLit-based web interface using a trained model
17
- - `dataset generation/generate_imgset1.py` our first iteration generating a dataset directly with MatPlotLib; these images look rough.
18
- - `dataset generation/generate_imgset2.py` our second iteration that generates nicer-looking images
19
- - `dataset generation/generate_imgset3.py` adds random simulated shadows overlaying generated images
20
- - `dataset generation/generate_rnn_imgset.py` generates individual images for each of 12 leads, for input into an RNN (rnn code currently fails to learn).
21
- - `dataset generation/automold.py` library with image augmentation code for adding shadows
22
- - `training/cnn_learner.py` trains and saves a cnn on generated images.
23
-
24
- Feel free to [email me](swow2015@mymail.pomona.edu) with any questions!