Long-Tail-ECG

Self-supervised Anomaly Detection Pretraining Enhances Long-tail ECG Diagnosis

Environment

numpy==1.22.4
pandas==1.1.3
Pillow==8.4.0
torch==1.10.1
scipy==1.7.3
heartpy==1.2.7

Checkpoint

Download the ckpt from https://drive.google.com/file/d/1E4NuKirypMk_Uy6iHenkUfP9cNEtvJi7/view?usp=sharing and put it under model_template/weights/

Run

run test.py for evaluation on the given data run train.py for training