torch_ecg icon indicating copy to clipboard operation
torch_ecg copied to clipboard

Training with peaks

Open mario-sanz opened this issue 2 years ago • 1 comments

Hello,

I would like to train a UNET network with the LUDB dataset but also with the peaks, not just the P, QRS and T waves (I would like also to detect P, R and T peaks). Is this possible? I have been doing lots of tests but I am not able to do so.

Thanks,

Mario

mario-sanz avatar Dec 15 '23 11:12 mario-sanz

Peaks are almost the "argmax" of corresponding waves, perhaps you can do a peak detection (sometimes valley detection) within the ranges of the waves after wave delineation (segmentation). Or you can narrow the segmentation labels to a small neighborhood of the peaks (e.g. ±20ms) and train a model using these labels, then the peaks would be the center of the segmentation result of the model inference.

wenh06 avatar Dec 16 '23 08:12 wenh06