|
a |
|
b/README.md |
|
|
1 |
来源:["合肥高新杯"心电人机智能大赛](https://tianchi.aliyun.com/competition/entrance/231754/introduction) |
|
|
2 |
|
|
|
3 |
**A版单模型20190916在线F1-Score=0.801** |
|
|
4 |
|
|
|
5 |
**B版单模型20191010在线F1-Score=0.837** |
|
|
6 |
|
|
|
7 |
系统环境:*centos7 python3.6 pytorch1.0* |
|
|
8 |
|
|
|
9 |
**大致思路:每个导联作为一个通道,使用1维卷积进行训练** |
|
|
10 |
|
|
|
11 |
# 数据预处理 |
|
|
12 |
数据解压放在data目录下,使用8个导联的数据,简单进行train_val数据集划分 |
|
|
13 |
```shell |
|
|
14 |
python data_process.py |
|
|
15 |
``` |
|
|
16 |
|
|
|
17 |
# 模型训练 |
|
|
18 |
```shell |
|
|
19 |
python main.py train #从零开始训练 |
|
|
20 |
``` |
|
|
21 |
|
|
|
22 |
# 模型测试 |
|
|
23 |
模型测试,在submit文件夹下生成提交结果 |
|
|
24 |
```shell |
|
|
25 |
python main.py test --ckpt=..model_path #加载预训练权重进行测试 |
|
|
26 |
``` |
|
|
27 |
|
|
|
28 |
**一些细节** |
|
|
29 |
|
|
|
30 |
1. 本次测试模型为1dconv_resnet34,直接修改于torchvision |
|
|
31 |
2. 训练数据只进行了简单的数据增强,最终无normalize |
|
|
32 |
3. 由于设备问题,数据进行了重采样,推荐使用原数据 |
|
|
33 |
|
|
|
34 |
|
|
|
35 |
参考论文: |
|
|
36 |
https://www.nature.com/articles/s41591-018-0268-3 |