├── 7Scenes_config.json ├── CambridgeLandmarks_config.json ├── CambridgeLandmarks_finetune_config.json ├── README.md ├── datasets ├── 7Scenes │ ├── 7scenes_all_scenes.csv │ ├── abs_7scenes_pose.csv_chess_test.csv │ ├── abs_7scenes_pose.csv_chess_train.csv │ ├── abs_7scenes_pose.csv_fire_test.csv │ ├── abs_7scenes_pose.csv_fire_train.csv │ ├── abs_7scenes_pose.csv_heads_test.csv │ ├── abs_7scenes_pose.csv_heads_train.csv │ ├── abs_7scenes_pose.csv_office_test.csv │ ├── abs_7scenes_pose.csv_office_train.csv │ ├── abs_7scenes_pose.csv_pumpkin_test.csv │ ├── abs_7scenes_pose.csv_pumpkin_train.csv │ ├── abs_7scenes_pose.csv_redkitchen_test.csv │ ├── abs_7scenes_pose.csv_redkitchen_train.csv │ ├── abs_7scenes_pose.csv_stairs_test.csv │ └── abs_7scenes_pose.csv_stairs_train.csv ├── CambridgeLandmarks │ ├── abs_cambridge_pose_sorted.csv_GreatCourt_test.csv │ ├── abs_cambridge_pose_sorted.csv_GreatCourt_train.csv │ ├── abs_cambridge_pose_sorted.csv_KingsCollege_test.csv │ ├── abs_cambridge_pose_sorted.csv_KingsCollege_train.csv │ ├── abs_cambridge_pose_sorted.csv_OldHospital_test.csv │ ├── abs_cambridge_pose_sorted.csv_OldHospital_train.csv │ ├── abs_cambridge_pose_sorted.csv_ShopFacade_test.csv │ ├── abs_cambridge_pose_sorted.csv_ShopFacade_train.csv │ ├── abs_cambridge_pose_sorted.csv_StMarysChurch_test.csv │ ├── abs_cambridge_pose_sorted.csv_StMarysChurch_train.csv │ ├── abs_cambridge_pose_sorted.csv_Street_test.csv │ ├── abs_cambridge_pose_sorted.csv_Street_train.csv │ ├── cambridge_five_scenes.csv │ └── cambridge_four_scenes.csv └── CameraPoseDataset.py ├── example_configs.json ├── img ├── cropped_teaser_small.pdf └── teaser.PNG ├── main.py ├── models ├── backbones │ └── efficient-net-b0.pth ├── pose_losses.py ├── pose_regressors.py ├── posenet │ └── PoseNet.py └── transposenet │ ├── EMSTransPoseNet.py │ ├── MSTransPoseNet.py │ ├── backbone.py │ ├── pencoder.py │ ├── transformer.py │ └── transformer_encoder.py ├── plots ├── ms-transformer-activation-maps-example.pickle └── plot_activation_maps.py ├── requirements.txt ├── scripts └── combine_scenes.py └── util ├── log_config.json └── utils.py /7Scenes_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/7Scenes_config.json -------------------------------------------------------------------------------- /CambridgeLandmarks_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/CambridgeLandmarks_config.json -------------------------------------------------------------------------------- /CambridgeLandmarks_finetune_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/CambridgeLandmarks_finetune_config.json -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/README.md -------------------------------------------------------------------------------- /datasets/7Scenes/7scenes_all_scenes.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/7Scenes/7scenes_all_scenes.csv -------------------------------------------------------------------------------- /datasets/7Scenes/abs_7scenes_pose.csv_chess_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/7Scenes/abs_7scenes_pose.csv_chess_test.csv -------------------------------------------------------------------------------- /datasets/7Scenes/abs_7scenes_pose.csv_chess_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/7Scenes/abs_7scenes_pose.csv_chess_train.csv -------------------------------------------------------------------------------- /datasets/7Scenes/abs_7scenes_pose.csv_fire_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/7Scenes/abs_7scenes_pose.csv_fire_test.csv -------------------------------------------------------------------------------- /datasets/7Scenes/abs_7scenes_pose.csv_fire_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/7Scenes/abs_7scenes_pose.csv_fire_train.csv -------------------------------------------------------------------------------- /datasets/7Scenes/abs_7scenes_pose.csv_heads_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/7Scenes/abs_7scenes_pose.csv_heads_test.csv -------------------------------------------------------------------------------- /datasets/7Scenes/abs_7scenes_pose.csv_heads_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/7Scenes/abs_7scenes_pose.csv_heads_train.csv -------------------------------------------------------------------------------- /datasets/7Scenes/abs_7scenes_pose.csv_office_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/7Scenes/abs_7scenes_pose.csv_office_test.csv -------------------------------------------------------------------------------- /datasets/7Scenes/abs_7scenes_pose.csv_office_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/7Scenes/abs_7scenes_pose.csv_office_train.csv -------------------------------------------------------------------------------- /datasets/7Scenes/abs_7scenes_pose.csv_pumpkin_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/7Scenes/abs_7scenes_pose.csv_pumpkin_test.csv -------------------------------------------------------------------------------- /datasets/7Scenes/abs_7scenes_pose.csv_pumpkin_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/7Scenes/abs_7scenes_pose.csv_pumpkin_train.csv -------------------------------------------------------------------------------- /datasets/7Scenes/abs_7scenes_pose.csv_redkitchen_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/7Scenes/abs_7scenes_pose.csv_redkitchen_test.csv -------------------------------------------------------------------------------- /datasets/7Scenes/abs_7scenes_pose.csv_redkitchen_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/7Scenes/abs_7scenes_pose.csv_redkitchen_train.csv -------------------------------------------------------------------------------- /datasets/7Scenes/abs_7scenes_pose.csv_stairs_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/7Scenes/abs_7scenes_pose.csv_stairs_test.csv -------------------------------------------------------------------------------- /datasets/7Scenes/abs_7scenes_pose.csv_stairs_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/7Scenes/abs_7scenes_pose.csv_stairs_train.csv -------------------------------------------------------------------------------- /datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_GreatCourt_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_GreatCourt_test.csv -------------------------------------------------------------------------------- /datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_GreatCourt_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_GreatCourt_train.csv -------------------------------------------------------------------------------- /datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_KingsCollege_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_KingsCollege_test.csv -------------------------------------------------------------------------------- /datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_KingsCollege_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_KingsCollege_train.csv -------------------------------------------------------------------------------- /datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_OldHospital_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_OldHospital_test.csv -------------------------------------------------------------------------------- /datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_OldHospital_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_OldHospital_train.csv -------------------------------------------------------------------------------- /datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_ShopFacade_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_ShopFacade_test.csv -------------------------------------------------------------------------------- /datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_ShopFacade_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_ShopFacade_train.csv -------------------------------------------------------------------------------- /datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_StMarysChurch_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_StMarysChurch_test.csv -------------------------------------------------------------------------------- /datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_StMarysChurch_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_StMarysChurch_train.csv -------------------------------------------------------------------------------- /datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_Street_test.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_Street_test.csv -------------------------------------------------------------------------------- /datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_Street_train.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/CambridgeLandmarks/abs_cambridge_pose_sorted.csv_Street_train.csv -------------------------------------------------------------------------------- /datasets/CambridgeLandmarks/cambridge_five_scenes.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/CambridgeLandmarks/cambridge_five_scenes.csv -------------------------------------------------------------------------------- /datasets/CambridgeLandmarks/cambridge_four_scenes.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/CambridgeLandmarks/cambridge_four_scenes.csv -------------------------------------------------------------------------------- /datasets/CameraPoseDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/datasets/CameraPoseDataset.py -------------------------------------------------------------------------------- /example_configs.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/example_configs.json -------------------------------------------------------------------------------- /img/cropped_teaser_small.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/img/cropped_teaser_small.pdf -------------------------------------------------------------------------------- /img/teaser.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/img/teaser.PNG -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/main.py -------------------------------------------------------------------------------- /models/backbones/efficient-net-b0.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/models/backbones/efficient-net-b0.pth -------------------------------------------------------------------------------- /models/pose_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/models/pose_losses.py -------------------------------------------------------------------------------- /models/pose_regressors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/models/pose_regressors.py -------------------------------------------------------------------------------- /models/posenet/PoseNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/models/posenet/PoseNet.py -------------------------------------------------------------------------------- /models/transposenet/EMSTransPoseNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/models/transposenet/EMSTransPoseNet.py -------------------------------------------------------------------------------- /models/transposenet/MSTransPoseNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/models/transposenet/MSTransPoseNet.py -------------------------------------------------------------------------------- /models/transposenet/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/models/transposenet/backbone.py -------------------------------------------------------------------------------- /models/transposenet/pencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/models/transposenet/pencoder.py -------------------------------------------------------------------------------- /models/transposenet/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/models/transposenet/transformer.py -------------------------------------------------------------------------------- /models/transposenet/transformer_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/models/transposenet/transformer_encoder.py -------------------------------------------------------------------------------- /plots/ms-transformer-activation-maps-example.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/plots/ms-transformer-activation-maps-example.pickle -------------------------------------------------------------------------------- /plots/plot_activation_maps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/plots/plot_activation_maps.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/combine_scenes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/scripts/combine_scenes.py -------------------------------------------------------------------------------- /util/log_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/util/log_config.json -------------------------------------------------------------------------------- /util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yolish/multi-scene-pose-transformer/HEAD/util/utils.py --------------------------------------------------------------------------------