Home

Awesome

ICIF-Net

Papers

image-model

1. Environment setup

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

pip install -r requirements.txt

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 ICIF-Net and DMINet in your work please cite our paper: