├── LICENSE ├── README.md ├── configs ├── _base_ │ ├── datasets │ │ ├── bdd_as_shift_track_det.py │ │ ├── bdd_pedestrian_track_det.py │ │ ├── dancetrack.py │ │ ├── dancetrack_from_bdd.py │ │ ├── mot_challenge.py │ │ ├── mot_challenge_from_bdd.py │ │ └── shift_track.py │ ├── default_runtime.py │ ├── models │ │ └── faster_rcnn_r50_fpn.py │ └── target_datasets │ │ ├── bdd_as_shift.py │ │ ├── bdd_pedestrian.py │ │ ├── bdd_pedestrian_from_mot.py │ │ ├── dancetrack.py │ │ ├── dancetrack_from_bdd.py │ │ ├── mot_challenge.py │ │ └── mot_challenge_from_bdd.py ├── adapters │ └── darth │ │ ├── darth_qdtrack_faster-rcnn_r50_fpn_12e_base.py │ │ ├── darth_qdtrack_faster-rcnn_r50_fpn_12e_bdd_as_shift.py │ │ ├── darth_qdtrack_faster-rcnn_r50_fpn_12e_bdd_pedestrian_from_mot.py │ │ ├── darth_qdtrack_faster-rcnn_r50_fpn_4e_dancetrack.py │ │ ├── darth_qdtrack_faster-rcnn_r50_fpn_4e_dancetrack_from_bdd.py │ │ ├── darth_qdtrack_faster-rcnn_r50_fpn_4e_mot17-private-half.py │ │ ├── darth_qdtrack_faster-rcnn_r50_fpn_4e_mot17-private-half_from_bdd.py │ │ └── darth_qdtrack_faster_rcnn_r50_fpn_4e_base.py └── mot │ └── qdtrack │ ├── qdtrack_faster-rcnn_r50_fpn_12e_base.py │ ├── qdtrack_faster-rcnn_r50_fpn_12e_bdd_as_shift_track_det.py │ ├── qdtrack_faster-rcnn_r50_fpn_12e_bdd_pedestrian_from_mot_track_det.py │ ├── qdtrack_faster-rcnn_r50_fpn_12e_bdd_pedestrian_track_det.py │ ├── qdtrack_faster-rcnn_r50_fpn_12e_bdd_track_det.py │ ├── qdtrack_faster-rcnn_r50_fpn_12e_dancetrack.py │ ├── qdtrack_faster-rcnn_r50_fpn_4e_base.py │ ├── qdtrack_faster-rcnn_r50_fpn_4e_crowdhuman_mot17-private-half.py │ ├── qdtrack_faster-rcnn_r50_fpn_4e_dancetrack.py │ ├── qdtrack_faster-rcnn_r50_fpn_4e_mot17-private-half.py │ ├── qdtrack_faster-rcnn_r50_fpn_4e_mot_challenge_from_bdd_pedestrian.py │ └── qdtrack_faster-rcnn_r50_fpn_6e_shift_track.py ├── darth ├── __init__.py ├── apis │ ├── __init__.py │ ├── inference.py │ ├── test.py │ └── train.py ├── core │ ├── __init__.py │ ├── hooks │ │ ├── __init__.py │ │ ├── ema.py │ │ └── optimizer.py │ ├── to_scalabel │ │ ├── __init__.py │ │ ├── transforms.py │ │ └── utils.py │ ├── track │ │ ├── __init__.py │ │ └── postprocessing.py │ └── utils │ │ ├── __init__.py │ │ └── visualization.py ├── datasets │ ├── __init__.py │ ├── bdd_tracking_dataset.py │ ├── builder.py │ ├── cat_mappings │ │ ├── bdd100k.toml │ │ ├── bdd100k_to_shift.toml │ │ └── waymo_to_bdd100k.toml │ ├── coco_tracking_dataset.py │ ├── dancetrack_tracking_dataset.py │ ├── dataset_wrappers.py │ ├── eval │ │ ├── __init__.py │ │ ├── coco_eval_dataset.py │ │ ├── coco_eval_utils.py │ │ ├── eval_mot.py │ │ └── logging_utils.py │ ├── mot_tracking_dataset.py │ ├── parsers │ │ ├── __init__.py │ │ ├── coco_parser.py │ │ └── coco_video_parser.py │ ├── pipelines │ │ ├── __init__.py │ │ ├── formatting.py │ │ ├── io │ │ │ ├── __init__.py │ │ │ └── tarbackend.py │ │ ├── loading.py │ │ └── transforms.py │ ├── shift_tracking_dataset.py │ └── utils │ │ ├── __init__.py │ │ ├── coco_to_scalabel.py │ │ └── filters.py ├── models │ ├── __init__.py │ ├── adapters │ │ ├── __init__.py │ │ ├── base │ │ │ ├── __init__.py │ │ │ └── teacher_qdtrack.py │ │ ├── darth_qdtrack.py │ │ └── kd_qdtrack.py │ ├── losses │ │ ├── __init__.py │ │ └── distillation.py │ ├── roi_heads │ │ ├── __init__.py │ │ └── modified_roi_head.py │ ├── trackers │ │ ├── __init__.py │ │ └── quasi_dense_tracker.py │ └── utils │ │ ├── __init__.py │ │ ├── kd_weight_init.py │ │ ├── transforms.py │ │ └── vis.py └── version.py ├── demo └── demo_mot.py ├── docs ├── DATASET.md ├── GET_STARTED.md ├── INSTALL.md └── SHIFT.md ├── resources └── darth_banner.png ├── scripts └── gdown_large.sh ├── setup.cfg ├── setup.py ├── test └── testcases │ └── bdd100k-sample │ ├── annotations │ └── box_track_debug_cocofmt.json │ ├── config.toml │ ├── images │ ├── b1c81faa-3df17267 │ │ ├── b1c81faa-3df17267-0000001.jpg │ │ ├── b1c81faa-3df17267-0000002.jpg │ │ ├── b1c81faa-3df17267-0000003.jpg │ │ ├── b1c81faa-3df17267-0000004.jpg │ │ ├── b1c81faa-3df17267-0000005.jpg │ │ └── b1c81faa-3df17267-0000006.jpg │ └── b1cac6a7-04e33135 │ │ ├── b1cac6a7-04e33135-0000001.jpg │ │ ├── b1cac6a7-04e33135-0000002.jpg │ │ ├── b1cac6a7-04e33135-0000003.jpg │ │ ├── b1cac6a7-04e33135-0000004.jpg │ │ ├── b1cac6a7-04e33135-0000005.jpg │ │ └── b1cac6a7-04e33135-0000006.jpg │ └── labels │ ├── b1c81faa-3df17267.json │ └── b1cac6a7-04e33135.json └── tools ├── convert_datasets ├── bdd100k │ ├── attributes_to_track.py │ └── location_to_det.py ├── dancetrack │ └── dancetrack2coco.py ├── mot │ ├── crowdhuman2coco.py │ ├── mot2coco.py │ └── mot2reid.py └── to_scalabel.py ├── install ├── install_gpu_conda.sh ├── install_gpu_venv.sh └── requirements.txt └── run ├── dist_test.sh ├── dist_train.sh ├── euler_dist_train.sh ├── euler_train.sh ├── resume.sh ├── sbatch_resume.sh ├── sbatch_test.sh ├── sbatch_train.sh ├── test.py ├── test.sh ├── train.py └── train.sh /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/README.md -------------------------------------------------------------------------------- /configs/_base_/datasets/bdd_as_shift_track_det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/datasets/bdd_as_shift_track_det.py -------------------------------------------------------------------------------- /configs/_base_/datasets/bdd_pedestrian_track_det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/datasets/bdd_pedestrian_track_det.py -------------------------------------------------------------------------------- /configs/_base_/datasets/dancetrack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/datasets/dancetrack.py -------------------------------------------------------------------------------- /configs/_base_/datasets/dancetrack_from_bdd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/datasets/dancetrack_from_bdd.py -------------------------------------------------------------------------------- /configs/_base_/datasets/mot_challenge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/datasets/mot_challenge.py -------------------------------------------------------------------------------- /configs/_base_/datasets/mot_challenge_from_bdd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/datasets/mot_challenge_from_bdd.py -------------------------------------------------------------------------------- /configs/_base_/datasets/shift_track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/datasets/shift_track.py -------------------------------------------------------------------------------- /configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/_base_/models/faster_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/models/faster_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /configs/_base_/target_datasets/bdd_as_shift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/target_datasets/bdd_as_shift.py -------------------------------------------------------------------------------- /configs/_base_/target_datasets/bdd_pedestrian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/target_datasets/bdd_pedestrian.py -------------------------------------------------------------------------------- /configs/_base_/target_datasets/bdd_pedestrian_from_mot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/target_datasets/bdd_pedestrian_from_mot.py -------------------------------------------------------------------------------- /configs/_base_/target_datasets/dancetrack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/target_datasets/dancetrack.py -------------------------------------------------------------------------------- /configs/_base_/target_datasets/dancetrack_from_bdd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/target_datasets/dancetrack_from_bdd.py -------------------------------------------------------------------------------- /configs/_base_/target_datasets/mot_challenge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/target_datasets/mot_challenge.py -------------------------------------------------------------------------------- /configs/_base_/target_datasets/mot_challenge_from_bdd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/_base_/target_datasets/mot_challenge_from_bdd.py -------------------------------------------------------------------------------- /configs/adapters/darth/darth_qdtrack_faster-rcnn_r50_fpn_12e_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/adapters/darth/darth_qdtrack_faster-rcnn_r50_fpn_12e_base.py -------------------------------------------------------------------------------- /configs/adapters/darth/darth_qdtrack_faster-rcnn_r50_fpn_12e_bdd_as_shift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/adapters/darth/darth_qdtrack_faster-rcnn_r50_fpn_12e_bdd_as_shift.py -------------------------------------------------------------------------------- /configs/adapters/darth/darth_qdtrack_faster-rcnn_r50_fpn_12e_bdd_pedestrian_from_mot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/adapters/darth/darth_qdtrack_faster-rcnn_r50_fpn_12e_bdd_pedestrian_from_mot.py -------------------------------------------------------------------------------- /configs/adapters/darth/darth_qdtrack_faster-rcnn_r50_fpn_4e_dancetrack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/adapters/darth/darth_qdtrack_faster-rcnn_r50_fpn_4e_dancetrack.py -------------------------------------------------------------------------------- /configs/adapters/darth/darth_qdtrack_faster-rcnn_r50_fpn_4e_dancetrack_from_bdd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/adapters/darth/darth_qdtrack_faster-rcnn_r50_fpn_4e_dancetrack_from_bdd.py -------------------------------------------------------------------------------- /configs/adapters/darth/darth_qdtrack_faster-rcnn_r50_fpn_4e_mot17-private-half.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/adapters/darth/darth_qdtrack_faster-rcnn_r50_fpn_4e_mot17-private-half.py -------------------------------------------------------------------------------- /configs/adapters/darth/darth_qdtrack_faster-rcnn_r50_fpn_4e_mot17-private-half_from_bdd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/adapters/darth/darth_qdtrack_faster-rcnn_r50_fpn_4e_mot17-private-half_from_bdd.py -------------------------------------------------------------------------------- /configs/adapters/darth/darth_qdtrack_faster_rcnn_r50_fpn_4e_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/adapters/darth/darth_qdtrack_faster_rcnn_r50_fpn_4e_base.py -------------------------------------------------------------------------------- /configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_12e_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_12e_base.py -------------------------------------------------------------------------------- /configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_12e_bdd_as_shift_track_det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_12e_bdd_as_shift_track_det.py -------------------------------------------------------------------------------- /configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_12e_bdd_pedestrian_from_mot_track_det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_12e_bdd_pedestrian_from_mot_track_det.py -------------------------------------------------------------------------------- /configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_12e_bdd_pedestrian_track_det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_12e_bdd_pedestrian_track_det.py -------------------------------------------------------------------------------- /configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_12e_bdd_track_det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_12e_bdd_track_det.py -------------------------------------------------------------------------------- /configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_12e_dancetrack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_12e_dancetrack.py -------------------------------------------------------------------------------- /configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_4e_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_4e_base.py -------------------------------------------------------------------------------- /configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_4e_crowdhuman_mot17-private-half.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_4e_crowdhuman_mot17-private-half.py -------------------------------------------------------------------------------- /configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_4e_dancetrack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_4e_dancetrack.py -------------------------------------------------------------------------------- /configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_4e_mot17-private-half.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_4e_mot17-private-half.py -------------------------------------------------------------------------------- /configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_4e_mot_challenge_from_bdd_pedestrian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_4e_mot_challenge_from_bdd_pedestrian.py -------------------------------------------------------------------------------- /configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_6e_shift_track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/configs/mot/qdtrack/qdtrack_faster-rcnn_r50_fpn_6e_shift_track.py -------------------------------------------------------------------------------- /darth/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/__init__.py -------------------------------------------------------------------------------- /darth/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/apis/__init__.py -------------------------------------------------------------------------------- /darth/apis/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/apis/inference.py -------------------------------------------------------------------------------- /darth/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/apis/test.py -------------------------------------------------------------------------------- /darth/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/apis/train.py -------------------------------------------------------------------------------- /darth/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/core/__init__.py -------------------------------------------------------------------------------- /darth/core/hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/core/hooks/__init__.py -------------------------------------------------------------------------------- /darth/core/hooks/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/core/hooks/ema.py -------------------------------------------------------------------------------- /darth/core/hooks/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/core/hooks/optimizer.py -------------------------------------------------------------------------------- /darth/core/to_scalabel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/core/to_scalabel/__init__.py -------------------------------------------------------------------------------- /darth/core/to_scalabel/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/core/to_scalabel/transforms.py -------------------------------------------------------------------------------- /darth/core/to_scalabel/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/core/to_scalabel/utils.py -------------------------------------------------------------------------------- /darth/core/track/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/core/track/__init__.py -------------------------------------------------------------------------------- /darth/core/track/postprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/core/track/postprocessing.py -------------------------------------------------------------------------------- /darth/core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/core/utils/__init__.py -------------------------------------------------------------------------------- /darth/core/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/core/utils/visualization.py -------------------------------------------------------------------------------- /darth/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/__init__.py -------------------------------------------------------------------------------- /darth/datasets/bdd_tracking_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/bdd_tracking_dataset.py -------------------------------------------------------------------------------- /darth/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/builder.py -------------------------------------------------------------------------------- /darth/datasets/cat_mappings/bdd100k.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/cat_mappings/bdd100k.toml -------------------------------------------------------------------------------- /darth/datasets/cat_mappings/bdd100k_to_shift.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/cat_mappings/bdd100k_to_shift.toml -------------------------------------------------------------------------------- /darth/datasets/cat_mappings/waymo_to_bdd100k.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/cat_mappings/waymo_to_bdd100k.toml -------------------------------------------------------------------------------- /darth/datasets/coco_tracking_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/coco_tracking_dataset.py -------------------------------------------------------------------------------- /darth/datasets/dancetrack_tracking_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/dancetrack_tracking_dataset.py -------------------------------------------------------------------------------- /darth/datasets/dataset_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/dataset_wrappers.py -------------------------------------------------------------------------------- /darth/datasets/eval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/eval/__init__.py -------------------------------------------------------------------------------- /darth/datasets/eval/coco_eval_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/eval/coco_eval_dataset.py -------------------------------------------------------------------------------- /darth/datasets/eval/coco_eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/eval/coco_eval_utils.py -------------------------------------------------------------------------------- /darth/datasets/eval/eval_mot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/eval/eval_mot.py -------------------------------------------------------------------------------- /darth/datasets/eval/logging_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/eval/logging_utils.py -------------------------------------------------------------------------------- /darth/datasets/mot_tracking_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/mot_tracking_dataset.py -------------------------------------------------------------------------------- /darth/datasets/parsers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/parsers/__init__.py -------------------------------------------------------------------------------- /darth/datasets/parsers/coco_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/parsers/coco_parser.py -------------------------------------------------------------------------------- /darth/datasets/parsers/coco_video_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/parsers/coco_video_parser.py -------------------------------------------------------------------------------- /darth/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /darth/datasets/pipelines/formatting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/pipelines/formatting.py -------------------------------------------------------------------------------- /darth/datasets/pipelines/io/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/pipelines/io/__init__.py -------------------------------------------------------------------------------- /darth/datasets/pipelines/io/tarbackend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/pipelines/io/tarbackend.py -------------------------------------------------------------------------------- /darth/datasets/pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/pipelines/loading.py -------------------------------------------------------------------------------- /darth/datasets/pipelines/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/pipelines/transforms.py -------------------------------------------------------------------------------- /darth/datasets/shift_tracking_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/shift_tracking_dataset.py -------------------------------------------------------------------------------- /darth/datasets/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/utils/__init__.py -------------------------------------------------------------------------------- /darth/datasets/utils/coco_to_scalabel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/utils/coco_to_scalabel.py -------------------------------------------------------------------------------- /darth/datasets/utils/filters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/datasets/utils/filters.py -------------------------------------------------------------------------------- /darth/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/__init__.py -------------------------------------------------------------------------------- /darth/models/adapters/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/adapters/__init__.py -------------------------------------------------------------------------------- /darth/models/adapters/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/adapters/base/__init__.py -------------------------------------------------------------------------------- /darth/models/adapters/base/teacher_qdtrack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/adapters/base/teacher_qdtrack.py -------------------------------------------------------------------------------- /darth/models/adapters/darth_qdtrack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/adapters/darth_qdtrack.py -------------------------------------------------------------------------------- /darth/models/adapters/kd_qdtrack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/adapters/kd_qdtrack.py -------------------------------------------------------------------------------- /darth/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/losses/__init__.py -------------------------------------------------------------------------------- /darth/models/losses/distillation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/losses/distillation.py -------------------------------------------------------------------------------- /darth/models/roi_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/roi_heads/__init__.py -------------------------------------------------------------------------------- /darth/models/roi_heads/modified_roi_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/roi_heads/modified_roi_head.py -------------------------------------------------------------------------------- /darth/models/trackers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/trackers/__init__.py -------------------------------------------------------------------------------- /darth/models/trackers/quasi_dense_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/trackers/quasi_dense_tracker.py -------------------------------------------------------------------------------- /darth/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/utils/__init__.py -------------------------------------------------------------------------------- /darth/models/utils/kd_weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/utils/kd_weight_init.py -------------------------------------------------------------------------------- /darth/models/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/utils/transforms.py -------------------------------------------------------------------------------- /darth/models/utils/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/models/utils/vis.py -------------------------------------------------------------------------------- /darth/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/darth/version.py -------------------------------------------------------------------------------- /demo/demo_mot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/demo/demo_mot.py -------------------------------------------------------------------------------- /docs/DATASET.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/docs/DATASET.md -------------------------------------------------------------------------------- /docs/GET_STARTED.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/docs/GET_STARTED.md -------------------------------------------------------------------------------- /docs/INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/docs/INSTALL.md -------------------------------------------------------------------------------- /docs/SHIFT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/docs/SHIFT.md -------------------------------------------------------------------------------- /resources/darth_banner.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/resources/darth_banner.png -------------------------------------------------------------------------------- /scripts/gdown_large.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/scripts/gdown_large.sh -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/setup.py -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/annotations/box_track_debug_cocofmt.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/annotations/box_track_debug_cocofmt.json -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/config.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/config.toml -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/images/b1c81faa-3df17267/b1c81faa-3df17267-0000001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/images/b1c81faa-3df17267/b1c81faa-3df17267-0000001.jpg -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/images/b1c81faa-3df17267/b1c81faa-3df17267-0000002.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/images/b1c81faa-3df17267/b1c81faa-3df17267-0000002.jpg -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/images/b1c81faa-3df17267/b1c81faa-3df17267-0000003.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/images/b1c81faa-3df17267/b1c81faa-3df17267-0000003.jpg -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/images/b1c81faa-3df17267/b1c81faa-3df17267-0000004.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/images/b1c81faa-3df17267/b1c81faa-3df17267-0000004.jpg -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/images/b1c81faa-3df17267/b1c81faa-3df17267-0000005.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/images/b1c81faa-3df17267/b1c81faa-3df17267-0000005.jpg -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/images/b1c81faa-3df17267/b1c81faa-3df17267-0000006.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/images/b1c81faa-3df17267/b1c81faa-3df17267-0000006.jpg -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/images/b1cac6a7-04e33135/b1cac6a7-04e33135-0000001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/images/b1cac6a7-04e33135/b1cac6a7-04e33135-0000001.jpg -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/images/b1cac6a7-04e33135/b1cac6a7-04e33135-0000002.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/images/b1cac6a7-04e33135/b1cac6a7-04e33135-0000002.jpg -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/images/b1cac6a7-04e33135/b1cac6a7-04e33135-0000003.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/images/b1cac6a7-04e33135/b1cac6a7-04e33135-0000003.jpg -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/images/b1cac6a7-04e33135/b1cac6a7-04e33135-0000004.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/images/b1cac6a7-04e33135/b1cac6a7-04e33135-0000004.jpg -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/images/b1cac6a7-04e33135/b1cac6a7-04e33135-0000005.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/images/b1cac6a7-04e33135/b1cac6a7-04e33135-0000005.jpg -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/images/b1cac6a7-04e33135/b1cac6a7-04e33135-0000006.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/images/b1cac6a7-04e33135/b1cac6a7-04e33135-0000006.jpg -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/labels/b1c81faa-3df17267.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/labels/b1c81faa-3df17267.json -------------------------------------------------------------------------------- /test/testcases/bdd100k-sample/labels/b1cac6a7-04e33135.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/test/testcases/bdd100k-sample/labels/b1cac6a7-04e33135.json -------------------------------------------------------------------------------- /tools/convert_datasets/bdd100k/attributes_to_track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/convert_datasets/bdd100k/attributes_to_track.py -------------------------------------------------------------------------------- /tools/convert_datasets/bdd100k/location_to_det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/convert_datasets/bdd100k/location_to_det.py -------------------------------------------------------------------------------- /tools/convert_datasets/dancetrack/dancetrack2coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/convert_datasets/dancetrack/dancetrack2coco.py -------------------------------------------------------------------------------- /tools/convert_datasets/mot/crowdhuman2coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/convert_datasets/mot/crowdhuman2coco.py -------------------------------------------------------------------------------- /tools/convert_datasets/mot/mot2coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/convert_datasets/mot/mot2coco.py -------------------------------------------------------------------------------- /tools/convert_datasets/mot/mot2reid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/convert_datasets/mot/mot2reid.py -------------------------------------------------------------------------------- /tools/convert_datasets/to_scalabel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/convert_datasets/to_scalabel.py -------------------------------------------------------------------------------- /tools/install/install_gpu_conda.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/install/install_gpu_conda.sh -------------------------------------------------------------------------------- /tools/install/install_gpu_venv.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/install/install_gpu_venv.sh -------------------------------------------------------------------------------- /tools/install/requirements.txt: -------------------------------------------------------------------------------- 1 | geopy 2 | h5py 3 | kornia 4 | pytest 5 | tensorboard 6 | wandb -------------------------------------------------------------------------------- /tools/run/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/run/dist_test.sh -------------------------------------------------------------------------------- /tools/run/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/run/dist_train.sh -------------------------------------------------------------------------------- /tools/run/euler_dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/run/euler_dist_train.sh -------------------------------------------------------------------------------- /tools/run/euler_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/run/euler_train.sh -------------------------------------------------------------------------------- /tools/run/resume.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/run/resume.sh -------------------------------------------------------------------------------- /tools/run/sbatch_resume.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/run/sbatch_resume.sh -------------------------------------------------------------------------------- /tools/run/sbatch_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/run/sbatch_test.sh -------------------------------------------------------------------------------- /tools/run/sbatch_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/run/sbatch_train.sh -------------------------------------------------------------------------------- /tools/run/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/run/test.py -------------------------------------------------------------------------------- /tools/run/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/run/test.sh -------------------------------------------------------------------------------- /tools/run/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/run/train.py -------------------------------------------------------------------------------- /tools/run/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mattiasegu/darth/HEAD/tools/run/train.sh --------------------------------------------------------------------------------