Skip to content
heartKIT
Reference
HELIA

2-Stage ECG Segmentation (SEG-2-TCN-SM)

The following table provides the latest pre-trained model for 2-stage ECG segmentation. Below we also provide additional details including training configuration, performance metrics, and downloads.

NAMEDATASETFSDURATION# CLASSESMODELPARAMSFLOPSMETRIC
SEG-2-TCN-SMLUDB, Synthetic100Hz2.5s2TCN2K0.42M96.6% F1

The model is trained on 2.5-second, raw ECG frames sampled at 100 Hz.

  • Sensor: ECG
  • Location: Wrist
  • Sampling Rate: 100 Hz
  • Frame Size: 2.5 seconds

The model is able to segment ECG signals into two classes: QRS complexes and none. The class mapping is as follows:

Base ClassTarget ClassLabel
0-NONE0NONE
2-QRS1QRS

The model is trained on the following datasets:


The confusion matrix for the model is depicted below.


AssetDescription
configuration.jsonConfiguration file
model.kerasKeras Model file
model.tfliteTFLite Model file
metrics.jsonMetrics file