CVPR 2024 "Unifying Top-down and Bottom-up Scanpath Prediction Using Transformers"
- Install customized Detectron2: Copy the folder at
/data/add_disk0/zbyang/projects/detectron2/
inbigrod
to your working directory and install it as followscd detectron2 pip install -e .
- Install MSDeformableAttn:
cd ./scanpath_prediction_all/sptransformer/pixel_decoder/ops sh make.sh
- Train a model with
python train_sptransformer.py --hparams ./configs/coco_search18_dense_SSL.json --dataset-root <dataset_root>
This repository contains code for scanpath prediction models for the following papers. Please cite if you use this code base.
@InProceedings{yang2024unify,
author = {Yang, Zhibo and Mondal, Sounak and Ahn, Seoyoung and Xue, Ruoyu and Zelinsky, Gregory and Hoai, Minh and Samaras, Dimitris},
title = {Unifying Top-down and Bottom-up Scanpath Prediction Using Transformers},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2024}
}