/MASDF-Net

Primary LanguagePython

MASDF-Net

This repo holds the code of "MASDF-Net: A multi-attention codec network with selective and dynamic fusion for skin lesion segmentation."

Framework Overview


Experiments

ISIC Skin Lesion Segmentation Challenge

GPUs of memory>=4G shall be sufficient for this experiment.

  1. Preparing necessary data:

    • downloading ISIC training, validation and testing data from the official site, put the unzipped data in ./data.
    • run process.py to preprocess all the data, which generates data_{train, val, test}.npy and mask_{train, val, test}.npy.

  2. Testing:

    • run test_isic.py --ckpt_path='snapshots/XXXX.pth'.

  3. Training:

    • run train_isic.py; you may also want to change the default saving path or other hparams as well.

Acknowledgement

We are very grateful for these excellent works PVT, Polyp-PVT, CCNet,TransFuse and PraNet, which have provided the basis for our framework.

Questions

Please drop an email to fujinghao@stu.scu.edu.cn