/MSHFFA

Multi-scale high-frequency feature augmentation

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

MSHFFA

Multi-scale high-frequency feature augmentation

Effect of MSHFFA on POCUS dataset

img_3.png

Effect of MSHFFA on BLU dataset

img_2.png

GradCam comparison

img_4.png

Coupled Training

POCUS dataset

The split training-test dataset can be obtained from here

Training MSHFFA-added model with POCUS dataset

Run train_POCUS_MSHFFA.py (Note to change project dir and save dir)

BLU dataset

The dataset can be obtained from here

Training MSHFFA-added model with BLU dataset

Run train_BLU_MSHFFA.py (Note to change project dir and save dir)

References

  1. https://github.com/lpj-github-io/MWCNNv2