- 28 Jun 2024: Open-sourced model and test code
- 19 Aug 2024: Released prediction map, pretrained model, backbone, and datasets
- Training source code will be released upon paper acceptance.
Item | Baidu Pan | Google Drive |
---|---|---|
Prediction Map | Link | Link |
Pretrained Model | Link | Link |
Backbone Model | Link | Link |
Datasets | Link | Link |
- Python 3.8.13
- others packages can be found at requirement.txt
-
conda create --name MAMIFNet python=3.8.12
-
conda activate MAMIFNet
-
pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
-
pip install -r requirements.txt
-
pip install -U openmim
-
mim install mmcv-full==1.5.2
The relevant setting paths for data and weights are located in config.py
. The default test is as follows:
-
create folder './data', download datasets file COD_dataset at https://pan.baidu.com/s/1Kpxaqv0n5YP-kDT8Id4wzA?pwd=pugv, unzip and move it to './data'
-
download backbone model at https://pan.baidu.com/s/16bKJaxewVfFXd73ICV6zDg?pwd=vn6v and unzip it.
-
download pretrained model at https://pan.baidu.com/s/10gWYyESaV6LaCZv8_1PVBA?pwd=z8d1 and unzip it.
-
run test.py