├── .gitignore ├── .idea └── .gitignore ├── README.md ├── configs └── Res_train.yaml ├── dataset ├── all_label_augment.json ├── all_label_map.json ├── cpsc │ └── data_preprocess.py ├── ecgDataset.py ├── georgia │ └── data_preprocess.py ├── label_augmentation.py ├── mimiciv │ ├── data_process.py │ ├── mimiciv_label_map_report.json │ ├── mlb.pkl │ └── total_label_set.csv ├── ptb-xl │ ├── data_preprocess.py │ └── report_translation.py └── shaoxing │ └── data_preprocess.py ├── engine ├── finetune_fg.py └── train_fg.py ├── factory ├── loss.py ├── metric.py ├── utils.py └── visualization.py ├── main.py ├── main_mimiciv.py ├── models ├── ECGNet.py ├── __init__.py ├── clip_model.py ├── cpc.py ├── model_new.py ├── resnet1d.py ├── resnet1d_wang.py ├── swinTransformer1d.py ├── swinTransformer1d_v2.py ├── transformer_decoder.py ├── vit_1d.py ├── xresnet1d_101.py └── xresnet1d_101_new.py ├── optim ├── __init__.py ├── adahessian.py ├── adamp.py ├── adamw.py ├── lookahead.py ├── nadam.py ├── novograd.py ├── nvnovograd.py ├── optim_factory.py ├── radam.py ├── rmsprop_tf.py └── sgdp.py ├── plot ├── ECG_plot.py └── plot_visualize.py ├── requirements.txt ├── scheduler ├── __init__.py ├── cosine_lr.py ├── plateau_lr.py ├── scheduler.py ├── scheduler_factory.py ├── step_lr.py └── tanh_lr.py ├── test_ICBEB.py ├── test_clinical.py ├── test_georgia.py ├── test_ptbxl.py └── test_shaoxing.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/.gitignore -------------------------------------------------------------------------------- /.idea/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/.idea/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/README.md -------------------------------------------------------------------------------- /configs/Res_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/configs/Res_train.yaml -------------------------------------------------------------------------------- /dataset/all_label_augment.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/dataset/all_label_augment.json -------------------------------------------------------------------------------- /dataset/all_label_map.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/dataset/all_label_map.json -------------------------------------------------------------------------------- /dataset/cpsc/data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/dataset/cpsc/data_preprocess.py -------------------------------------------------------------------------------- /dataset/ecgDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/dataset/ecgDataset.py -------------------------------------------------------------------------------- /dataset/georgia/data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/dataset/georgia/data_preprocess.py -------------------------------------------------------------------------------- /dataset/label_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/dataset/label_augmentation.py -------------------------------------------------------------------------------- /dataset/mimiciv/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/dataset/mimiciv/data_process.py -------------------------------------------------------------------------------- /dataset/mimiciv/mimiciv_label_map_report.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/dataset/mimiciv/mimiciv_label_map_report.json -------------------------------------------------------------------------------- /dataset/mimiciv/mlb.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/dataset/mimiciv/mlb.pkl -------------------------------------------------------------------------------- /dataset/mimiciv/total_label_set.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/dataset/mimiciv/total_label_set.csv -------------------------------------------------------------------------------- /dataset/ptb-xl/data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/dataset/ptb-xl/data_preprocess.py -------------------------------------------------------------------------------- /dataset/ptb-xl/report_translation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/dataset/ptb-xl/report_translation.py -------------------------------------------------------------------------------- /dataset/shaoxing/data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/dataset/shaoxing/data_preprocess.py -------------------------------------------------------------------------------- /engine/finetune_fg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/engine/finetune_fg.py -------------------------------------------------------------------------------- /engine/train_fg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/engine/train_fg.py -------------------------------------------------------------------------------- /factory/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/factory/loss.py -------------------------------------------------------------------------------- /factory/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/factory/metric.py -------------------------------------------------------------------------------- /factory/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/factory/utils.py -------------------------------------------------------------------------------- /factory/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/factory/visualization.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/main.py -------------------------------------------------------------------------------- /main_mimiciv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/main_mimiciv.py -------------------------------------------------------------------------------- /models/ECGNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/models/ECGNet.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/clip_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/models/clip_model.py -------------------------------------------------------------------------------- /models/cpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/models/cpc.py -------------------------------------------------------------------------------- /models/model_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/models/model_new.py -------------------------------------------------------------------------------- /models/resnet1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/models/resnet1d.py -------------------------------------------------------------------------------- /models/resnet1d_wang.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/models/resnet1d_wang.py -------------------------------------------------------------------------------- /models/swinTransformer1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/models/swinTransformer1d.py -------------------------------------------------------------------------------- /models/swinTransformer1d_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/models/swinTransformer1d_v2.py -------------------------------------------------------------------------------- /models/transformer_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/models/transformer_decoder.py -------------------------------------------------------------------------------- /models/vit_1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/models/vit_1d.py -------------------------------------------------------------------------------- /models/xresnet1d_101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/models/xresnet1d_101.py -------------------------------------------------------------------------------- /models/xresnet1d_101_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/models/xresnet1d_101_new.py -------------------------------------------------------------------------------- /optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/optim/__init__.py -------------------------------------------------------------------------------- /optim/adahessian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/optim/adahessian.py -------------------------------------------------------------------------------- /optim/adamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/optim/adamp.py -------------------------------------------------------------------------------- /optim/adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/optim/adamw.py -------------------------------------------------------------------------------- /optim/lookahead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/optim/lookahead.py -------------------------------------------------------------------------------- /optim/nadam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/optim/nadam.py -------------------------------------------------------------------------------- /optim/novograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/optim/novograd.py -------------------------------------------------------------------------------- /optim/nvnovograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/optim/nvnovograd.py -------------------------------------------------------------------------------- /optim/optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/optim/optim_factory.py -------------------------------------------------------------------------------- /optim/radam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/optim/radam.py -------------------------------------------------------------------------------- /optim/rmsprop_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/optim/rmsprop_tf.py -------------------------------------------------------------------------------- /optim/sgdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/optim/sgdp.py -------------------------------------------------------------------------------- /plot/ECG_plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/plot/ECG_plot.py -------------------------------------------------------------------------------- /plot/plot_visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/plot/plot_visualize.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/requirements.txt -------------------------------------------------------------------------------- /scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/scheduler/__init__.py -------------------------------------------------------------------------------- /scheduler/cosine_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/scheduler/cosine_lr.py -------------------------------------------------------------------------------- /scheduler/plateau_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/scheduler/plateau_lr.py -------------------------------------------------------------------------------- /scheduler/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/scheduler/scheduler.py -------------------------------------------------------------------------------- /scheduler/scheduler_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/scheduler/scheduler_factory.py -------------------------------------------------------------------------------- /scheduler/step_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/scheduler/step_lr.py -------------------------------------------------------------------------------- /scheduler/tanh_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/scheduler/tanh_lr.py -------------------------------------------------------------------------------- /test_ICBEB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/test_ICBEB.py -------------------------------------------------------------------------------- /test_clinical.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/test_clinical.py -------------------------------------------------------------------------------- /test_georgia.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/test_georgia.py -------------------------------------------------------------------------------- /test_ptbxl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/test_ptbxl.py -------------------------------------------------------------------------------- /test_shaoxing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/control-spiderman/ECGFM-KED/HEAD/test_shaoxing.py --------------------------------------------------------------------------------