scene recognition with attention
-
use utility/make_hha_encode/encode_sun_rgbd.py to generate hha image, which is time costly.
-
use nyud2_meta/make_common_10.py to generate meta data of the dataset;
python work.py --lr 3e-4 --l2 0.5 --log_root /path/to/tensorboard_log/ --epochs 10000 --pretrain_dir /path/to/alexnet_places365.pth.tar --save_inteval_epoch 1 --device cuda:0