Preprocessing-Method-for-STEMI-Detection

KO | EN

gitlite — search

#python #ai #typescript #llm #android #machine-learning #nodejs #macos #windows #cli #awesome-list #mcp #openai #open-source

Preprocessing-Method-for-STEMI-Detection

(★ 16)

Official source code of "Preprocessing Method for Performance Enhancement in CNN-based STEMI Detection from 12-lead ECG"

Preprocessing-Method-for-STEMI-Detection Latest Version Download

Download Latest Version (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

Preprocessing Method for Performance Enhancement in CNN-based STEMI Detection from 12-lead ECG
=====

This repository provides the source code of the paper "Preprocessing Method for Performance Enhancement in CNN-based STEMI Detection from 12-lead ECG" [<a href="https://ieeexplore.ieee.org/stamp/stamp.jsp?tp=&arnumber=8771175">pdf</a>].

## STEMI: ST-elevation myocardial infarction
The complete occlusion of the coronary artery is called STEMI and it is characterized by a sudden shut-down of blood flow caused by thrombus or embolism. In the ECG of STEMI, the ST-segment is more elevated than the normal.

<div align="center">
  <img src="./figures/normal.png" width="500">  
  <img src="./figures/stemi.png"  width="500">  
  <p>The ECG of normal (upper) and STEMI (lower)</p>
</div>

## Requirements
* Python 3.5.2  
* Tensorflow 1.4.0  
* Numpy 1.13.3  
* Scipy 1.2.0  
* WFDB 2.2.1  
* Matplotlib 3.0.2

## Usage
### Preparing the dataset
First, Organize the dataset and keep as below. Keep the channel (or lead) information in axis 0, and time information in axis 1. For example, 12 lead data with 5500-time length (500Hz x 11 seconds) should be saved as `(12, 5500)` form.  
```
Raw_ECG
├── Normal
│   ├── data_1
│   ├── data_2
│   ├── data_3
│   │     ...
│   └── data_n
└── STEMI
```
Then, run the python script as following.  
```
$ python preprocessing.py
$ python preprocessing.py --help // for confirming the option
```

In the process of executing the above source code, pulse segmentation is performed by voting and choosing the location of the QRS complex as shown below.

<div align="center">
  <img src="./figures/qrs_voted.png"  width="500">  
  <img src="./figures/qrs_lead_i.png" width="500">  
  <img src="./figures/qrs_lead_avr.png"  width="500">  
  <p>The top of the figure shows location voting result for finding the QRS complex. Lower two figure shows most voted time location as QRS complex at lead-I and lead-aVR respectively.</p>
</div>

### Training and Test
```
$ cd CNN
$ python run_resnet.py
```
Using `run_vggnet.py` instead of `run_resnet.py` is available for training VGGNet.

<div align="center">
  <img src="./figures/vggnet16.png" width="800">  
  <p>1D-VGGNet-16</p>  
  <img src="./figures/resnet34.png"  width="800">  
  <p>1D-ResNet-34</p>
</div>

### Pre-Trained CNN
If you want to use pre-trained model it is available at <a href="https://drive.google.com/drive/folders/1qexMFed1SIO2n1R5EE2QoxCaEA1iiDbf?usp=sharing">Google Drive</a>. Saved parameters are provided for 1D-VGGNet-16 and 1D-ResNet-34 respectively.

Available since Mar.31.2019

### BibTeX
```
@Article{8771175,
  author={Park, YeongHyeon and Yun, Il Dong and Kang, Si-Hyuck},
  journal={IEEE Access},
  title={Preprocessing Method for Performance Enhancement in CNN-Based STEMI Detection From 12-Lead ECG},
  year={2019},
  volume={7},
  pages={99964-99977},
  ISSN={2169-3536},
}
```

Preprocessing-Method-for-STEMI-Detection

Preprocessing-Method-for-STEMI-Detection Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Preprocessing-Method-for-STEMI-Detection

Preprocessing-Method-for-STEMI-Detection Latest Version Download

Today's picks

Was this content helpful?

Recent Feedback

Download README