[发明专利]模型训练方法、装置、电子设备及存储介质

说明书

本发明实施例公开了一种模型训练方法、装置、电子设备及存储介质，其中，该方法包括：获取样本训练数据集，其中，样本训练数据集中包括至少一个样本心电信号数据以及与样本心电信号数据对应的样本心跳峰值标注数据，各个样本心电信号数据对应的信号采集通道的通道数量相同；将样本训练数据集中的样本心电信号数据输入至初始心跳峰值检测模型中，得到模型心跳峰值标注数据；根据模型心跳峰值标注数据和样本心跳峰值标注数据对初始心跳峰值检测模型的模型参数进行调整，以得到目标心跳峰值检测模型。本发明实施例的技术方案，实现了可以对不同通道类型的心电信号数据进行心跳峰值检测的效果，减轻了工作负担，提升了用户体验。

技术领域

本发明实施例涉及医疗技术领域，尤其涉及一种模型训练方法、装置、电子设备及存储介质。

背景技术

心电图(Electro Cardio Gram，ECG)心跳峰值检测是心电分析的一个重要功能，为心跳计数、心律变异性、波峰值高度计算以及根据心跳进行信号分段等工作奠定基础。除了医用的标准导联的ECG数据，现在各类家用、日常使用的ECG测量设备越来越多。这些设备一般会有各具特色的电极(通道)设置，其通道的数量、采集位置以及在记录数据中的通道顺序各不相同。

现有技术中，心跳峰值检测方法需要人工在考虑不同通道数、不同通道顺序以及不同采集位置的基础上，对ECG数据进行处理，不仅只能对应特定的情况，还增加了工作的重复冗余。

发明内容

本发明实施例提供了一种模型训练方法、装置、电子设备及存储介质，以实现适应不同类型的通道配置的心电信号的心跳峰值检测。

第一方面，本发明实施例提供了一种模型训练方法，该方法包括：

获取样本训练数据集，其中，样本训练数据集中包括至少一个样本心电信号数据以及与样本心电信号数据对应的样本心跳峰值标注数据，各个样本心电信号数据对应的信号采集通道的通道数量相同；

将样本训练数据集中的样本心电信号数据输入至初始心跳峰值检测模型中，得到模型心跳峰值标注数据；

根据模型心跳峰值标注数据和样本心跳峰值标注数据对初始心跳峰值检测模型的模型参数进行调整，以得到目标心跳峰值检测模型。

第二方面，本发明实施例还提供了一种模型训练装置，该装置包括：

数据集获取模块，用于获取样本训练数据集，其中，样本训练数据集中包括至少一个样本心电信号数据以及与样本心电信号数据对应的样本心跳峰值标注数据，各个样本心电信号数据对应的信号采集通道的通道数量相同；

数据输入模块，用于将样本训练数据集中的样本心电信号数据输入至初始心跳峰值检测模型中，得到模型心跳峰值标注数据；

模型参数调整模块，用于根据模型心跳峰值标注数据和样本心跳峰值标注数据对初始心跳峰值检测模型的模型参数进行调整，以得到目标心跳峰值检测模型。

第三方面，本发明实施例还提供了一种电子设备，该电子设备包括：

一个或多个处理器；

存储装置，用于存储一个或多个程序，

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现本发明任意实施例所提供的模型训练方法。

第四方面，本发明实施例还提供了一种计算机可读存储介质，其上存储有计算机程序，该计算机程序被处理器执行时实现本发明任意实施例所提供的模型训练方法。