├── Figures ├── .keep └── arch.png ├── LICENSE ├── README.md ├── TranSalNet_Dense.py ├── TranSalNet_Res.py ├── datasets ├── train │ ├── train_fixation │ │ └── .keep │ ├── train_saliency │ │ └── .keep │ └── train_stimuli │ │ └── .keep ├── train_ids.csv ├── val │ ├── val_fixation │ │ └── .keep │ ├── val_saliency │ │ └── .keep │ └── val_stimuli │ │ └── .keep └── val_ids.csv ├── example ├── COCO_val2014_000000005107.jpg ├── result_DensNet.png └── result_ResNet.png ├── fine-tune&train.ipynb ├── pretrained_models └── .keep ├── testing.ipynb └── utils ├── TransformerEncoder.py ├── data_process.py ├── densenet.py ├── loss_function.py └── resnet.py /Figures/.keep: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /Figures/arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/Figures/arch.png -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/README.md -------------------------------------------------------------------------------- /TranSalNet_Dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/TranSalNet_Dense.py -------------------------------------------------------------------------------- /TranSalNet_Res.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/TranSalNet_Res.py -------------------------------------------------------------------------------- /datasets/train/train_fixation/.keep: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /datasets/train/train_saliency/.keep: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /datasets/train/train_stimuli/.keep: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /datasets/train_ids.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/datasets/train_ids.csv -------------------------------------------------------------------------------- /datasets/val/val_fixation/.keep: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /datasets/val/val_saliency/.keep: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /datasets/val/val_stimuli/.keep: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /datasets/val_ids.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/datasets/val_ids.csv -------------------------------------------------------------------------------- /example/COCO_val2014_000000005107.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/example/COCO_val2014_000000005107.jpg -------------------------------------------------------------------------------- /example/result_DensNet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/example/result_DensNet.png -------------------------------------------------------------------------------- /example/result_ResNet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/example/result_ResNet.png -------------------------------------------------------------------------------- /fine-tune&train.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/fine-tune&train.ipynb -------------------------------------------------------------------------------- /pretrained_models/.keep: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /testing.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/testing.ipynb -------------------------------------------------------------------------------- /utils/TransformerEncoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/utils/TransformerEncoder.py -------------------------------------------------------------------------------- /utils/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/utils/data_process.py -------------------------------------------------------------------------------- /utils/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/utils/densenet.py -------------------------------------------------------------------------------- /utils/loss_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/utils/loss_function.py -------------------------------------------------------------------------------- /utils/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LJOVO/TranSalNet/HEAD/utils/resnet.py --------------------------------------------------------------------------------