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

Name		Name	Last commit message	Last commit date
Latest commit History 3 Commits
dataset		dataset
symbols		symbols
utils		utils
.gitignore		.gitignore
README.md		README.md
config.yml		config.yml
eval.py		eval.py
finetune.py		finetune.py
finetune_v2.py		finetune_v2.py
rel_scene.py		rel_scene.py
test.jpg		test.jpg
test2.jpg		test2.jpg
train.sh		train.sh
train_ls.sh		train_ls.sh
train_mit67_ls.sh		train_mit67_ls.sh
train_mit67_nonlocal_ls.sh		train_mit67_nonlocal_ls.sh
train_sun397.sh		train_sun397.sh
train_sun397_nonlocal_ls.sh		train_sun397_nonlocal_ls.sh
train_sun397_nonlocal_ls_v2.sh		train_sun397_nonlocal_ls_v2.sh
train_tmp.sh		train_tmp.sh

ChengpengChen/scene-attention