Home

Awesome

DMINet

Papers

1. Environment setup

This code has been tested on on the workstation with Intel Xeon CPU E5-2690 v4 cores and GPUs of NVIDIA TITAN V with a single 12G of video memory, Python 3.6, pytorch 1.9, CUDA 10.0, cuDNN 7.6.

2. Download the datesets:

and put them into data directory.

3. Download the models (loading models):

and put them into checkpoints directory.

4. Train

python main_cd.py

5. Test

python eval_cd.py

6. Cite

If you use DMINet in your work please cite our paper:

If you use ICIF-Net in your work please cite our paper: