用于心电疾病诊断的深度学习模型库
github: https://github.com/hzzhangqf0558/ECG_Nets
Baseline model collection of deep learning applied into ECGs. Those baseline models include 1D-ResNet, 1D-DenseNet, 1D-SE_ResNet, 1D-ResNext,1D-SE_ResNetV2, 1D-SE_ResNext and 1D-Top1Net(the champion model in Tianchi competition).
The F1-scores of those baseline models are in range of 0.83-0.90. you could fine-tune the parameters to reach a better level.
Step 1:
downloading datasets. The datasets from Tianchi competition includes dataset A and dataset B. The dataset A consists of 24106 ECGs, of which each has 8 leading records(I,II,V1,V2,V3,V4,V5, V6). while the dataset B consists of 20036 ECGs. Dataset can be downloaded from website.
https://pan.baidu.com/s/1fmCuV5i9oifnUNOsFhV0sA pwd: 8hs2
Please put the dataset