Skip to content
heartKIT
Reference
HELIA

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

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

NAMEDATASETFSDURATION# CLASSESMODELPARAMSFLOPSMETRIC
SEG-4-TCN-SMLUDB, Synthetic100Hz2.5s4TCN7K2.1M86.3% F1

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

Identify each of the P-wave, QRS complex, and T-wave.

Base ClassTarget ClassLabel
0-NONE0NONE
1-PWAVE1PWAVE
2-QRS2QRS
3-TWAVE3TWAVE

The model is trained on the following datasets:


The confusion matrix for the segmentation model is depicted below.


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