Pytorch Implementation of DUCK-Net for polyp image segmentation

Pytorch Implementation of DUCK-Net

Original code is available in github and you can see detail of this model in paper.

Installation

Requirements

• Python >= 3.10
• Pytorch >= 2.2.0
• CUDA 12.0

It's expected to work for latest versions too.

git clone https://github.com/russel0719/DUCK-Net-Pytorch.git
pip install torch torchvision pillow tqdm

Train

1. Fix config directly or Fix codes in train.py under if __name__ == "__main__" statement.
2. Run train.py

cd DUCK-Net-Pytorch
python train.py

Predict

1. Fix config directly and Fix paths in predict.py under if __name__ == "__main__" statement.
2. Run predict.py

cd DUCK-Net-Pytorch
python predict.py

Reference

• Paper

  • Using DUCK-Net for polyp image segmentation

• Original Github

  • Tensorflow Implementation of RazvanDu