Skip to content
/ PCNet Public

Alignment-Free RGB-T Salient Object Detection: A Large-scale Dataset and Progressive Correlation Network

Notifications You must be signed in to change notification settings

Angknpng/PCNet

Repository files navigation

PCNet

Code and dataset repository for our paper entilted "Alignment-Free RGB-T Salient Object Detection: A Large-scale Dataset and Progressive Correlation Network" accepted at AAAI 2025.

arXiv version: https://arxiv.org/pdf/2412.14576.

The model and results are available now. [17th, Jul, 2025]

Thank you for your attention.

Dataset

avatar

The compressed UVT20K dataset containing the annotations of saliency maps, edges, scribbles, and challenge attributes can be found here. [baidu pan fetch code: v2rc] or [google drive]

Method

avatar

Results

The predicted results of our models can be found here. [baidu pan fetch code: nhau]

The parameters of our models can be found here. [baidu pan fetch code: bz6x]

The predicted results of the comparison methods can be found here. [baidu pan fetch code: 3kru]

Usage

Requirement

  1. Download the UVT20K dataset for training and testing.
  2. Download the pretrained parameters of the backbone from here. [baidu pan fetch code: mad3]
  3. Download the pretrained parameters of the IHN model from here.
  4. Organize dataset and pretrained model directories.
  5. Create directories for the experiment and parameter files.
  6. Please use conda to install torch (1.12.0) and torchvision (0.13.0).
  7. Install other packages: pip install -r requirements.txt.
  8. Set your path of all datasets in ./options.py.

Train

python -m torch.distributed.launch --nproc_per_node=2 --master_port=2212 train_parallel.py

Test

python test_produce_maps.py

Citation

If you think our work is helpful, please cite:

@inproceedings{wang2025alignment,
  title={Alignment-Free RGB-T Salient Object Detection: A Large-scale Dataset and Progressive Correlation Network},
  author={Wang, Kunpeng and Chen, Keke and Li, Chenglong and Tu, Zhengzheng and Luo, Bin},
  booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
  volume={39},
  number={7},
  pages={7780--7788},
  year={2025}
}

Acknowledgement

The implement of this project is based on the following link.

Contact

If you have any questions, please contact us ([email protected]).

About

Alignment-Free RGB-T Salient Object Detection: A Large-scale Dataset and Progressive Correlation Network

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages