Skip to content

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

Overview

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.

NAME DATASET FS DURATION # CLASSES MODEL PARAMS FLOPS METRIC
SEG-4-TCN-SM LUDB, Synthetic 100Hz 2.5s 4 TCN 7K 2.1M 86.3% F1

Input

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

Class Mapping

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

Base Class Target Class Label
0-NONE 0 NONE
1-PWAVE 1 PWAVE
2-QRS 2 QRS
3-TWAVE 3 TWAVE

Datasets

The model is trained on the following datasets:


Model Performance

The confusion matrix for the segmentation model is depicted below.


Downloads

Asset Description
configuration.json Configuration file
model.keras Keras Model file
model.tflite TFLite Model file
metrics.json Metrics file