Delete README.md
Browse files
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!
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|