上传者: 42099087
|
上传时间: 2021-10-20 09:54:01
|
文件大小: 22KB
|
文件类型: -
变形神经网络
使用pytorch进行心电图诊断的可变形卷积网络。 这是文章“基于具有良好抗噪能力的可变形卷积神经网络的端到端12导心电图诊断系统”的源代码
入门
训练环境
NVIDIA驱动程序版本418.67
CUDA版本10.1
Python版本3.6.8
Pytorch版本1.4.0 + cu101
安装所需的软件包
从该存储库的根目录运行
pip install -r requirements.txt
下载数据集
我们使用CPSC-2018作为数据集。 您可以从下载数据重新处理数据并以.npy格式保存数据。 然后将数据放入文件夹./dataset/DataSet250HzRepeatFill并将标签放入文件夹./dataset 。 有关更多详细信息,请参考load_dataset.py 。
注意:在我们的实验中,我们删除了总长度超过30秒的数据,将采样频率降低到250 Hz