/MCCANet

Primary LanguagePython

MCCANet

Papers

1. Environment setup

This code has been tested on on the workstation with Intel Xeon Silver 4210R CPU and GPUs of NVIDIA GeForce RTX 3090 with a single 24G of video memory, Python 3.8, pytorch 1.11.

2. Download the datesets:

and modify the dataset directory in the validation code.

3. Download the models (loading models):

and put it into log directory.

4. Train

coming soon

5. Test

python eval.py

6. Cite