心电信号分割，言简意赅的说法就是识别心电信号的P Q R S T。心电信号的R峰检测是一个热门课题，但是P Q R S T检测做的人相对没那么多。

1、心电信号分割数据集

常用的心电信号分割数据集包括QTDB和LUDB，以LUDB数据集为例，该数据集由 200 条 10 秒钟 12 导联心电信号组成。所有 200 条记录的 P 波、T 波和 QRS 波群的边界均由心脏病专家手动标注，可以使用wfdb.rdann函数用于读取注释。

2、U-Net模型

U-Net最先应用在医学图像上，当然，对于心电信号，U-Net同样适用。

模型的输入为一维心电信号以及mask后的标签，通过最小化CrossEntropyLoss损失，可以实现对心电信号的P Q R S T的分割。

3、心电信号分割效果

3.1 模型原始输出

3.2 带后处理的模型输出

可以看出，通过后处理后，很好地处理了模型输出的一些边界，使得预测效果更佳。

4、后话

心电信号分割在心电信号领域是一个比较小众的研究方向，并无什么难点。

如果有对心电信号分割不是很明白的同学，也欢迎联系我。