Feature Shrinkage Pyramid for Camouflaged Object Detection with Transformers (CVPR2023)

Usage

The training and testing experiments are conducted using PyTorch with 8 Tesla V100 GPUs of 36 GB Memory.

1. Prerequisites

Note that FSPNet is only tested on Ubuntu OS with the following environments.

Creating a virtual environment in terminal: conda create -n FSPNet python=3.8.
Installing necessary packages: pip install -r requirements.txt

2. Downloading Training and Testing Datasets

Download the training set (COD10K-train) used for training
Download the testing sets (COD10K-test + CAMO-test + CHAMELEON + NC4K ) used for testing

3. Training Configuration

The pretrained model is stored in Google Drive and Baidu Drive (xuwb). After downloading, please change the file path in the corresponding code.
Run train.sh or slurm_train.sh as needed to train.

4. Testing Configuration

Our well-trained model is stored in Google Drive and Baidu Drive (otz5). After downloading, please change the file path in the corresponding code.

5. Evaluation

Matlab code: One-key evaluation is written in MATLAB code, please follow this the instructions in main.m and just run it to generate the evaluation results.
Python code: After configuring the test dataset path, run slurm_eval.py in the run_slurm folder for evaluation.

6. Results download

The prediction results of our FSPNet are stored on Google Drive and Baidu Drive (ryzg) please check.

Citation

@inproceedings{Huang2023Feature,
title={Feature Shrinkage Pyramid for Camouflaged Object Detection with Transformers},
author={Huang, Zhou and Dai, Hang and Xiang, Tian-Zhu and Wang, Shuo and Chen, Huai-Xin and Qin, Jie and Xiong, Huan},
booktitle={IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
year={2023}
}

Thanks to Deng-Ping Fan, Ge-Peng Ji, et al. for a series of efforts in the field of COD.

ZhouHuang23 / FSPNet

readme