/AttSwinUNet-For-Echocardiogram-Segmentation

AttSwinUNet For LV Segmentation in Echocardiograms (The dataset used is CAMUS)

Primary LanguagePython

AttSwinUNet For Echocardiogram Segmentation

Implementation code for Attention Swin U-Net: Cross-Contextual Attention Mechanism for Skin Lesion Segmentation paper for segmentation of LV in Echocardiograms. The dataset used is CAMUS

Environment and Installation

  • Please prepare an environment with python=3.7, and then use the command below to install all the dependencies pip install -r requirements.txt

Running the code

Training

 python train.py

Validation

python eval.py
  • Modify hyperparameters in the code itself

References

Citation

@article{aghdam2022attention,
  title={Attention Swin U-Net: Cross-Contextual Attention Mechanism for Skin Lesion Segmentation},
  author={Aghdam, Ehsan Khodapanah and Azad, Reza and Zarvani, Maral and Merhof, Dorit},
  journal={arXiv preprint arXiv:2210.16898},
  year={2022}
}