├── .gitignore ├── LICENSE ├── README.md ├── assets ├── MOT17-04.gif └── MOT17-11.gif ├── configs ├── r50_deformable_detr.sh ├── r50_deformable_detr_plus_iterative_bbox_refinement.sh └── r50_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage.sh ├── datasets ├── __init__.py ├── coco.py ├── coco_eval.py ├── coco_panoptic.py ├── crowdhuman.py ├── data_prefetcher.py ├── mix.py ├── mot.py ├── panoptic_eval.py ├── sampler_video_distributed.py ├── samplers.py ├── torchvision_datasets │ ├── __init__.py │ └── coco.py └── transforms.py ├── demo.py ├── engine.py ├── engine_track.py ├── main.py ├── main_track.py ├── models ├── __init__.py ├── backbone.py ├── deformable_detr.py ├── deformable_detrtrack_test.py ├── deformable_detrtrack_train.py ├── deformable_transformer.py ├── deformable_transformer_track.py ├── matcher.py ├── ops │ ├── functions │ │ ├── __init__.py │ │ └── ms_deform_attn_func.py │ ├── make.sh │ ├── modules │ │ ├── __init__.py │ │ └── ms_deform_attn.py │ ├── setup.py │ ├── src │ │ ├── cpu │ │ │ ├── ms_deform_attn_cpu.cpp │ │ │ └── ms_deform_attn_cpu.h │ │ ├── cuda │ │ │ ├── ms_deform_attn_cuda.cu │ │ │ ├── ms_deform_attn_cuda.h │ │ │ └── ms_deform_im2col_cuda.cuh │ │ ├── ms_deform_attn.h │ │ └── vision.cpp │ ├── test.py │ └── test_double_precision.py ├── position_encoding.py ├── reid │ ├── deformable_detrtrack_test.py │ ├── deformable_detrtrack_train.py │ ├── deformable_transformer_track.py │ ├── save_track.py │ └── tracker.py ├── save_track.py ├── segmentation.py └── tracker.py ├── requirements.txt ├── track_exps ├── crowdhuman_mot_train.sh ├── crowdhuman_mot_trainhalf.sh ├── crowdhuman_plus_mot_train.sh ├── crowdhuman_train.sh ├── mot_test.sh ├── mot_trainhalf.sh ├── mot_val.sh ├── mot_val_mn.sh └── mota.sh ├── track_tools ├── colormap.py ├── convert_crowdhuman_to_coco.py ├── convert_mot_to_coco.py ├── crowdhuman_model_to_mot.py ├── eval_motchallenge.py ├── mix_data.py ├── show_ann.py └── txt2video.py ├── transtrack.png └── util ├── __init__.py ├── box_ops.py ├── misc.py └── plot_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/README.md -------------------------------------------------------------------------------- /assets/MOT17-04.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/assets/MOT17-04.gif -------------------------------------------------------------------------------- /assets/MOT17-11.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/assets/MOT17-11.gif -------------------------------------------------------------------------------- /configs/r50_deformable_detr.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/configs/r50_deformable_detr.sh -------------------------------------------------------------------------------- /configs/r50_deformable_detr_plus_iterative_bbox_refinement.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/configs/r50_deformable_detr_plus_iterative_bbox_refinement.sh -------------------------------------------------------------------------------- /configs/r50_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/configs/r50_deformable_detr_plus_iterative_bbox_refinement_plus_plus_two_stage.sh -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/datasets/coco.py -------------------------------------------------------------------------------- /datasets/coco_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/datasets/coco_eval.py -------------------------------------------------------------------------------- /datasets/coco_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/datasets/coco_panoptic.py -------------------------------------------------------------------------------- /datasets/crowdhuman.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/datasets/crowdhuman.py -------------------------------------------------------------------------------- /datasets/data_prefetcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/datasets/data_prefetcher.py -------------------------------------------------------------------------------- /datasets/mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/datasets/mix.py -------------------------------------------------------------------------------- /datasets/mot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/datasets/mot.py -------------------------------------------------------------------------------- /datasets/panoptic_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/datasets/panoptic_eval.py -------------------------------------------------------------------------------- /datasets/sampler_video_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/datasets/sampler_video_distributed.py -------------------------------------------------------------------------------- /datasets/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/datasets/samplers.py -------------------------------------------------------------------------------- /datasets/torchvision_datasets/__init__.py: -------------------------------------------------------------------------------- 1 | from .coco import CocoDetection 2 | -------------------------------------------------------------------------------- /datasets/torchvision_datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/datasets/torchvision_datasets/coco.py -------------------------------------------------------------------------------- /datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/datasets/transforms.py -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/demo.py -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/engine.py -------------------------------------------------------------------------------- /engine_track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/engine_track.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/main.py -------------------------------------------------------------------------------- /main_track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/main_track.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/backbone.py -------------------------------------------------------------------------------- /models/deformable_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/deformable_detr.py -------------------------------------------------------------------------------- /models/deformable_detrtrack_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/deformable_detrtrack_test.py -------------------------------------------------------------------------------- /models/deformable_detrtrack_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/deformable_detrtrack_train.py -------------------------------------------------------------------------------- /models/deformable_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/deformable_transformer.py -------------------------------------------------------------------------------- /models/deformable_transformer_track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/deformable_transformer_track.py -------------------------------------------------------------------------------- /models/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/matcher.py -------------------------------------------------------------------------------- /models/ops/functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/ops/functions/__init__.py -------------------------------------------------------------------------------- /models/ops/functions/ms_deform_attn_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/ops/functions/ms_deform_attn_func.py -------------------------------------------------------------------------------- /models/ops/make.sh: -------------------------------------------------------------------------------- 1 | python setup.py build install 2 | -------------------------------------------------------------------------------- /models/ops/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/ops/modules/__init__.py -------------------------------------------------------------------------------- /models/ops/modules/ms_deform_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/ops/modules/ms_deform_attn.py -------------------------------------------------------------------------------- /models/ops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/ops/setup.py -------------------------------------------------------------------------------- /models/ops/src/cpu/ms_deform_attn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/ops/src/cpu/ms_deform_attn_cpu.cpp -------------------------------------------------------------------------------- /models/ops/src/cpu/ms_deform_attn_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/ops/src/cpu/ms_deform_attn_cpu.h -------------------------------------------------------------------------------- /models/ops/src/cuda/ms_deform_attn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/ops/src/cuda/ms_deform_attn_cuda.cu -------------------------------------------------------------------------------- /models/ops/src/cuda/ms_deform_attn_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/ops/src/cuda/ms_deform_attn_cuda.h -------------------------------------------------------------------------------- /models/ops/src/cuda/ms_deform_im2col_cuda.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/ops/src/cuda/ms_deform_im2col_cuda.cuh -------------------------------------------------------------------------------- /models/ops/src/ms_deform_attn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/ops/src/ms_deform_attn.h -------------------------------------------------------------------------------- /models/ops/src/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/ops/src/vision.cpp -------------------------------------------------------------------------------- /models/ops/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/ops/test.py -------------------------------------------------------------------------------- /models/ops/test_double_precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/ops/test_double_precision.py -------------------------------------------------------------------------------- /models/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/position_encoding.py -------------------------------------------------------------------------------- /models/reid/deformable_detrtrack_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/reid/deformable_detrtrack_test.py -------------------------------------------------------------------------------- /models/reid/deformable_detrtrack_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/reid/deformable_detrtrack_train.py -------------------------------------------------------------------------------- /models/reid/deformable_transformer_track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/reid/deformable_transformer_track.py -------------------------------------------------------------------------------- /models/reid/save_track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/reid/save_track.py -------------------------------------------------------------------------------- /models/reid/tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/reid/tracker.py -------------------------------------------------------------------------------- /models/save_track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/save_track.py -------------------------------------------------------------------------------- /models/segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/segmentation.py -------------------------------------------------------------------------------- /models/tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/models/tracker.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | pycocotools 2 | tqdm 3 | cython 4 | scipy 5 | motmetrics -------------------------------------------------------------------------------- /track_exps/crowdhuman_mot_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_exps/crowdhuman_mot_train.sh -------------------------------------------------------------------------------- /track_exps/crowdhuman_mot_trainhalf.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_exps/crowdhuman_mot_trainhalf.sh -------------------------------------------------------------------------------- /track_exps/crowdhuman_plus_mot_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_exps/crowdhuman_plus_mot_train.sh -------------------------------------------------------------------------------- /track_exps/crowdhuman_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_exps/crowdhuman_train.sh -------------------------------------------------------------------------------- /track_exps/mot_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_exps/mot_test.sh -------------------------------------------------------------------------------- /track_exps/mot_trainhalf.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_exps/mot_trainhalf.sh -------------------------------------------------------------------------------- /track_exps/mot_val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_exps/mot_val.sh -------------------------------------------------------------------------------- /track_exps/mot_val_mn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_exps/mot_val_mn.sh -------------------------------------------------------------------------------- /track_exps/mota.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_exps/mota.sh -------------------------------------------------------------------------------- /track_tools/colormap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_tools/colormap.py -------------------------------------------------------------------------------- /track_tools/convert_crowdhuman_to_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_tools/convert_crowdhuman_to_coco.py -------------------------------------------------------------------------------- /track_tools/convert_mot_to_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_tools/convert_mot_to_coco.py -------------------------------------------------------------------------------- /track_tools/crowdhuman_model_to_mot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_tools/crowdhuman_model_to_mot.py -------------------------------------------------------------------------------- /track_tools/eval_motchallenge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_tools/eval_motchallenge.py -------------------------------------------------------------------------------- /track_tools/mix_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_tools/mix_data.py -------------------------------------------------------------------------------- /track_tools/show_ann.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_tools/show_ann.py -------------------------------------------------------------------------------- /track_tools/txt2video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/track_tools/txt2video.py -------------------------------------------------------------------------------- /transtrack.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/transtrack.png -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved 2 | -------------------------------------------------------------------------------- /util/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/util/box_ops.py -------------------------------------------------------------------------------- /util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/util/misc.py -------------------------------------------------------------------------------- /util/plot_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PeizeSun/TransTrack/HEAD/util/plot_utils.py --------------------------------------------------------------------------------