├── .gitignore ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── INSTALL.md ├── LICENSE ├── README.md ├── checkpoint └── train_log │ └── README.md ├── configs ├── Base-RCNN-FPN.yaml ├── siam_rcnn_2_gpus.yaml ├── siam_rcnn_8_gpus.yaml └── siam_rcnn_8_gpus_e2e_125k.yaml ├── convert_videos_to_clips.py ├── convert_videos_to_images.py ├── demo ├── ours_easy_frying_pan.gif └── ours_hard_blue_bin.gif ├── detectron2_extensions ├── __init__.py ├── config │ ├── __init__.py │ └── defaults.py ├── layers │ ├── __init__.py │ └── wrappers.py └── modeling │ ├── meta_arch │ ├── __init__.py │ └── siam_rcnn.py │ └── roi_heads │ ├── __init__.py │ ├── set_heads.py │ └── siam_heads.py ├── evaluate_vq2d.py ├── evaluate_vq2d_one_query.py ├── get_test_challenge_predictions.py ├── process_vq_dataset.py ├── requirements.txt ├── scripts ├── evaluate_vq.sh ├── faster_evaluation │ ├── merge_results.py │ ├── merge_results_test.py │ ├── slurm_eval_array.sh │ └── slurm_test_array.sh ├── get_challenge_predictions.sh ├── train_2_gpus.sh └── train_8_gpus.sh ├── slurm_8node_launch.py ├── train_siam_rcnn.py ├── validate_challenge_predictions.py └── vq2d ├── baselines ├── __init__.py ├── dataloader.py ├── dataset.py ├── feature_retrieval.py ├── predictor.py └── utils.py ├── config.yaml ├── constants.py ├── metrics ├── __init__.py ├── metrics.py ├── spatio_temporal_metrics.py ├── success_metrics.py ├── temporal_metrics.py ├── tracking_metrics.py └── utils.py ├── stats.py ├── structures.py └── tracking ├── __init__.py ├── kys.py ├── particle_filter.py ├── pfilter.py ├── tracker.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/.gitignore -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/INSTALL.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/README.md -------------------------------------------------------------------------------- /checkpoint/train_log/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/checkpoint/train_log/README.md -------------------------------------------------------------------------------- /configs/Base-RCNN-FPN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/configs/Base-RCNN-FPN.yaml -------------------------------------------------------------------------------- /configs/siam_rcnn_2_gpus.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/configs/siam_rcnn_2_gpus.yaml -------------------------------------------------------------------------------- /configs/siam_rcnn_8_gpus.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/configs/siam_rcnn_8_gpus.yaml -------------------------------------------------------------------------------- /configs/siam_rcnn_8_gpus_e2e_125k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/configs/siam_rcnn_8_gpus_e2e_125k.yaml -------------------------------------------------------------------------------- /convert_videos_to_clips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/convert_videos_to_clips.py -------------------------------------------------------------------------------- /convert_videos_to_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/convert_videos_to_images.py -------------------------------------------------------------------------------- /demo/ours_easy_frying_pan.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/demo/ours_easy_frying_pan.gif -------------------------------------------------------------------------------- /demo/ours_hard_blue_bin.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/demo/ours_hard_blue_bin.gif -------------------------------------------------------------------------------- /detectron2_extensions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/detectron2_extensions/__init__.py -------------------------------------------------------------------------------- /detectron2_extensions/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/detectron2_extensions/config/__init__.py -------------------------------------------------------------------------------- /detectron2_extensions/config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/detectron2_extensions/config/defaults.py -------------------------------------------------------------------------------- /detectron2_extensions/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/detectron2_extensions/layers/__init__.py -------------------------------------------------------------------------------- /detectron2_extensions/layers/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/detectron2_extensions/layers/wrappers.py -------------------------------------------------------------------------------- /detectron2_extensions/modeling/meta_arch/__init__.py: -------------------------------------------------------------------------------- 1 | from .siam_rcnn import SiameseRCNN 2 | -------------------------------------------------------------------------------- /detectron2_extensions/modeling/meta_arch/siam_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/detectron2_extensions/modeling/meta_arch/siam_rcnn.py -------------------------------------------------------------------------------- /detectron2_extensions/modeling/roi_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/detectron2_extensions/modeling/roi_heads/__init__.py -------------------------------------------------------------------------------- /detectron2_extensions/modeling/roi_heads/set_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/detectron2_extensions/modeling/roi_heads/set_heads.py -------------------------------------------------------------------------------- /detectron2_extensions/modeling/roi_heads/siam_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/detectron2_extensions/modeling/roi_heads/siam_heads.py -------------------------------------------------------------------------------- /evaluate_vq2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/evaluate_vq2d.py -------------------------------------------------------------------------------- /evaluate_vq2d_one_query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/evaluate_vq2d_one_query.py -------------------------------------------------------------------------------- /get_test_challenge_predictions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/get_test_challenge_predictions.py -------------------------------------------------------------------------------- /process_vq_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/process_vq_dataset.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/evaluate_vq.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/scripts/evaluate_vq.sh -------------------------------------------------------------------------------- /scripts/faster_evaluation/merge_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/scripts/faster_evaluation/merge_results.py -------------------------------------------------------------------------------- /scripts/faster_evaluation/merge_results_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/scripts/faster_evaluation/merge_results_test.py -------------------------------------------------------------------------------- /scripts/faster_evaluation/slurm_eval_array.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/scripts/faster_evaluation/slurm_eval_array.sh -------------------------------------------------------------------------------- /scripts/faster_evaluation/slurm_test_array.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/scripts/faster_evaluation/slurm_test_array.sh -------------------------------------------------------------------------------- /scripts/get_challenge_predictions.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/scripts/get_challenge_predictions.sh -------------------------------------------------------------------------------- /scripts/train_2_gpus.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/scripts/train_2_gpus.sh -------------------------------------------------------------------------------- /scripts/train_8_gpus.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/scripts/train_8_gpus.sh -------------------------------------------------------------------------------- /slurm_8node_launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/slurm_8node_launch.py -------------------------------------------------------------------------------- /train_siam_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/train_siam_rcnn.py -------------------------------------------------------------------------------- /validate_challenge_predictions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/validate_challenge_predictions.py -------------------------------------------------------------------------------- /vq2d/baselines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/baselines/__init__.py -------------------------------------------------------------------------------- /vq2d/baselines/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/baselines/dataloader.py -------------------------------------------------------------------------------- /vq2d/baselines/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/baselines/dataset.py -------------------------------------------------------------------------------- /vq2d/baselines/feature_retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/baselines/feature_retrieval.py -------------------------------------------------------------------------------- /vq2d/baselines/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/baselines/predictor.py -------------------------------------------------------------------------------- /vq2d/baselines/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/baselines/utils.py -------------------------------------------------------------------------------- /vq2d/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/config.yaml -------------------------------------------------------------------------------- /vq2d/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/constants.py -------------------------------------------------------------------------------- /vq2d/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/metrics/__init__.py -------------------------------------------------------------------------------- /vq2d/metrics/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/metrics/metrics.py -------------------------------------------------------------------------------- /vq2d/metrics/spatio_temporal_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/metrics/spatio_temporal_metrics.py -------------------------------------------------------------------------------- /vq2d/metrics/success_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/metrics/success_metrics.py -------------------------------------------------------------------------------- /vq2d/metrics/temporal_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/metrics/temporal_metrics.py -------------------------------------------------------------------------------- /vq2d/metrics/tracking_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/metrics/tracking_metrics.py -------------------------------------------------------------------------------- /vq2d/metrics/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/metrics/utils.py -------------------------------------------------------------------------------- /vq2d/stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/stats.py -------------------------------------------------------------------------------- /vq2d/structures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/structures.py -------------------------------------------------------------------------------- /vq2d/tracking/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/tracking/__init__.py -------------------------------------------------------------------------------- /vq2d/tracking/kys.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/tracking/kys.py -------------------------------------------------------------------------------- /vq2d/tracking/particle_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/tracking/particle_filter.py -------------------------------------------------------------------------------- /vq2d/tracking/pfilter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/tracking/pfilter.py -------------------------------------------------------------------------------- /vq2d/tracking/tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/tracking/tracker.py -------------------------------------------------------------------------------- /vq2d/tracking/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/vq2d_cvpr/HEAD/vq2d/tracking/utils.py --------------------------------------------------------------------------------