├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── configs ├── _base_ │ ├── faster_rcnn_r50_fpn.py │ └── qdtrack_faster_rcnn_r50_fpn.py ├── bdd100k │ ├── qdtrack-frcnn_r50_fpn_12e_bdd100k.py │ └── qdtrack-frcnn_r50_fpn_12e_bdd100k_search.py ├── mot17 │ ├── README.md │ └── qdtrack-frcnn_r50_fpn_4e_mot17.py └── tao │ ├── README.md │ ├── qdtrack_frcnn_r101_fpn_12e_tao_ft.py │ └── qdtrack_frcnn_r101_fpn_24e_lvis.py ├── demo ├── demo.mp4 └── output.mp4 ├── docs ├── GET_STARTED.md └── INSTALL.md ├── figures └── teaser.png ├── qdtrack ├── VERSION ├── __init__.py ├── apis │ ├── __init__.py │ ├── inference.py │ ├── test.py │ └── train.py ├── core │ ├── __init__.py │ ├── evaluation │ │ ├── __init__.py │ │ ├── eval_hooks.py │ │ └── mot.py │ ├── to_bdd100k │ │ ├── __init__.py │ │ ├── transforms.py │ │ └── utils.py │ ├── track │ │ ├── __init__.py │ │ ├── similarity.py │ │ └── transforms.py │ └── utils │ │ ├── __init__.py │ │ └── visualization.py ├── datasets │ ├── __init__.py │ ├── bdd_video_dataset.py │ ├── builder.py │ ├── coco_video_dataset.py │ ├── mot17_dataset.py │ ├── parsers │ │ ├── __init__.py │ │ ├── coco_api.py │ │ └── coco_video_parser.py │ ├── pipelines │ │ ├── __init__.py │ │ ├── formatting.py │ │ ├── loading.py │ │ └── transforms.py │ ├── samplers │ │ ├── __init__.py │ │ └── distributed_video_sampler.py │ └── tao_dataset.py ├── models │ ├── __init__.py │ ├── builder.py │ ├── losses │ │ ├── __init__.py │ │ ├── l2_loss.py │ │ └── multipos_cross_entropy_loss.py │ ├── mot │ │ ├── __init__.py │ │ └── qdtrack.py │ ├── roi_heads │ │ ├── __init__.py │ │ ├── quasi_dense_roi_head.py │ │ └── track_heads │ │ │ ├── __init__.py │ │ │ └── quasi_dense_embed_head.py │ └── trackers │ │ ├── __init__.py │ │ ├── quasi_dense_embed_tracker.py │ │ └── tao_tracker.py ├── utils │ ├── __init__.py │ ├── collect_env.py │ └── logger.py └── version.py ├── requirements.txt ├── setup.cfg ├── setup.py └── tools ├── convert_datasets ├── mot2coco.py └── tao2coco.py ├── dist_test.sh ├── dist_test_search.sh ├── dist_train.sh ├── inference.py ├── mot_param_search.py ├── slurm_test.sh ├── slurm_train.sh ├── test.py ├── to_bdd100k.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/README.md -------------------------------------------------------------------------------- /configs/_base_/faster_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/configs/_base_/faster_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /configs/_base_/qdtrack_faster_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/configs/_base_/qdtrack_faster_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /configs/bdd100k/qdtrack-frcnn_r50_fpn_12e_bdd100k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/configs/bdd100k/qdtrack-frcnn_r50_fpn_12e_bdd100k.py -------------------------------------------------------------------------------- /configs/bdd100k/qdtrack-frcnn_r50_fpn_12e_bdd100k_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/configs/bdd100k/qdtrack-frcnn_r50_fpn_12e_bdd100k_search.py -------------------------------------------------------------------------------- /configs/mot17/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/configs/mot17/README.md -------------------------------------------------------------------------------- /configs/mot17/qdtrack-frcnn_r50_fpn_4e_mot17.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/configs/mot17/qdtrack-frcnn_r50_fpn_4e_mot17.py -------------------------------------------------------------------------------- /configs/tao/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/configs/tao/README.md -------------------------------------------------------------------------------- /configs/tao/qdtrack_frcnn_r101_fpn_12e_tao_ft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/configs/tao/qdtrack_frcnn_r101_fpn_12e_tao_ft.py -------------------------------------------------------------------------------- /configs/tao/qdtrack_frcnn_r101_fpn_24e_lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/configs/tao/qdtrack_frcnn_r101_fpn_24e_lvis.py -------------------------------------------------------------------------------- /demo/demo.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/demo/demo.mp4 -------------------------------------------------------------------------------- /demo/output.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/demo/output.mp4 -------------------------------------------------------------------------------- /docs/GET_STARTED.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/docs/GET_STARTED.md -------------------------------------------------------------------------------- /docs/INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/docs/INSTALL.md -------------------------------------------------------------------------------- /figures/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/figures/teaser.png -------------------------------------------------------------------------------- /qdtrack/VERSION: -------------------------------------------------------------------------------- 1 | 0.1.0 2 | -------------------------------------------------------------------------------- /qdtrack/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/__init__.py -------------------------------------------------------------------------------- /qdtrack/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/apis/__init__.py -------------------------------------------------------------------------------- /qdtrack/apis/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/apis/inference.py -------------------------------------------------------------------------------- /qdtrack/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/apis/test.py -------------------------------------------------------------------------------- /qdtrack/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/apis/train.py -------------------------------------------------------------------------------- /qdtrack/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/core/__init__.py -------------------------------------------------------------------------------- /qdtrack/core/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/core/evaluation/__init__.py -------------------------------------------------------------------------------- /qdtrack/core/evaluation/eval_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/core/evaluation/eval_hooks.py -------------------------------------------------------------------------------- /qdtrack/core/evaluation/mot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/core/evaluation/mot.py -------------------------------------------------------------------------------- /qdtrack/core/to_bdd100k/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/core/to_bdd100k/__init__.py -------------------------------------------------------------------------------- /qdtrack/core/to_bdd100k/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/core/to_bdd100k/transforms.py -------------------------------------------------------------------------------- /qdtrack/core/to_bdd100k/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/core/to_bdd100k/utils.py -------------------------------------------------------------------------------- /qdtrack/core/track/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/core/track/__init__.py -------------------------------------------------------------------------------- /qdtrack/core/track/similarity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/core/track/similarity.py -------------------------------------------------------------------------------- /qdtrack/core/track/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/core/track/transforms.py -------------------------------------------------------------------------------- /qdtrack/core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/core/utils/__init__.py -------------------------------------------------------------------------------- /qdtrack/core/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/core/utils/visualization.py -------------------------------------------------------------------------------- /qdtrack/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/datasets/__init__.py -------------------------------------------------------------------------------- /qdtrack/datasets/bdd_video_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/datasets/bdd_video_dataset.py -------------------------------------------------------------------------------- /qdtrack/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/datasets/builder.py -------------------------------------------------------------------------------- /qdtrack/datasets/coco_video_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/datasets/coco_video_dataset.py -------------------------------------------------------------------------------- /qdtrack/datasets/mot17_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/datasets/mot17_dataset.py -------------------------------------------------------------------------------- /qdtrack/datasets/parsers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/datasets/parsers/__init__.py -------------------------------------------------------------------------------- /qdtrack/datasets/parsers/coco_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/datasets/parsers/coco_api.py -------------------------------------------------------------------------------- /qdtrack/datasets/parsers/coco_video_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/datasets/parsers/coco_video_parser.py -------------------------------------------------------------------------------- /qdtrack/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /qdtrack/datasets/pipelines/formatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/datasets/pipelines/formatting.py -------------------------------------------------------------------------------- /qdtrack/datasets/pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/datasets/pipelines/loading.py -------------------------------------------------------------------------------- /qdtrack/datasets/pipelines/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/datasets/pipelines/transforms.py -------------------------------------------------------------------------------- /qdtrack/datasets/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/datasets/samplers/__init__.py -------------------------------------------------------------------------------- /qdtrack/datasets/samplers/distributed_video_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/datasets/samplers/distributed_video_sampler.py -------------------------------------------------------------------------------- /qdtrack/datasets/tao_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/datasets/tao_dataset.py -------------------------------------------------------------------------------- /qdtrack/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/models/__init__.py -------------------------------------------------------------------------------- /qdtrack/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/models/builder.py -------------------------------------------------------------------------------- /qdtrack/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/models/losses/__init__.py -------------------------------------------------------------------------------- /qdtrack/models/losses/l2_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/models/losses/l2_loss.py -------------------------------------------------------------------------------- /qdtrack/models/losses/multipos_cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/models/losses/multipos_cross_entropy_loss.py -------------------------------------------------------------------------------- /qdtrack/models/mot/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/models/mot/__init__.py -------------------------------------------------------------------------------- /qdtrack/models/mot/qdtrack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/models/mot/qdtrack.py -------------------------------------------------------------------------------- /qdtrack/models/roi_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/models/roi_heads/__init__.py -------------------------------------------------------------------------------- /qdtrack/models/roi_heads/quasi_dense_roi_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/models/roi_heads/quasi_dense_roi_head.py -------------------------------------------------------------------------------- /qdtrack/models/roi_heads/track_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/models/roi_heads/track_heads/__init__.py -------------------------------------------------------------------------------- /qdtrack/models/roi_heads/track_heads/quasi_dense_embed_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/models/roi_heads/track_heads/quasi_dense_embed_head.py -------------------------------------------------------------------------------- /qdtrack/models/trackers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/models/trackers/__init__.py -------------------------------------------------------------------------------- /qdtrack/models/trackers/quasi_dense_embed_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/models/trackers/quasi_dense_embed_tracker.py -------------------------------------------------------------------------------- /qdtrack/models/trackers/tao_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/models/trackers/tao_tracker.py -------------------------------------------------------------------------------- /qdtrack/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/utils/__init__.py -------------------------------------------------------------------------------- /qdtrack/utils/collect_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/utils/collect_env.py -------------------------------------------------------------------------------- /qdtrack/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/utils/logger.py -------------------------------------------------------------------------------- /qdtrack/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/qdtrack/version.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/setup.py -------------------------------------------------------------------------------- /tools/convert_datasets/mot2coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/tools/convert_datasets/mot2coco.py -------------------------------------------------------------------------------- /tools/convert_datasets/tao2coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/tools/convert_datasets/tao2coco.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_test_search.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/tools/dist_test_search.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/tools/inference.py -------------------------------------------------------------------------------- /tools/mot_param_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/tools/mot_param_search.py -------------------------------------------------------------------------------- /tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/tools/slurm_test.sh -------------------------------------------------------------------------------- /tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/tools/slurm_train.sh -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/to_bdd100k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/tools/to_bdd100k.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SysCV/qdtrack/HEAD/tools/train.py --------------------------------------------------------------------------------