/scene-attention

scene attention on MIT67 and SUN397

Primary LanguagePython

scene-attention

Repo for scene attention on MIT67 and SUN397 dataset.

In process...

To run a baseline based on Resnet 50, you can refer to the following shell:

python finetune_v2.py --num_non_local_block 0 --dropout_ratio 0.5 --prefix baseline_resnet50_mit67_bs16_drop0.5