Pytorch Implementation of DUCK-Net
Original code is available in github and you can see detail of this model in paper.
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
- Fix config directly or Fix codes in train.py under
if __name__ == "__main__"
statement. - Run train.py
cd DUCK-Net-Pytorch
python train.py
- Fix config directly and Fix paths in predict.py under
if __name__ == "__main__"
statement. - Run predict.py
cd DUCK-Net-Pytorch
python predict.py
-
Paper
-
Original Github