├── LICENSE ├── README.md ├── demo └── pipeline.jpg ├── experiments └── train_stcmot.sh ├── models └── ctdet_coco_dla_2x.pth ├── requirements.txt └── src ├── __pycache__ ├── _init_paths.cpython-38.pyc └── gen_dataset_visdrone.cpython-38.pyc ├── _init_paths.py ├── data ├── UAVDT.test ├── UAVDT.train ├── VisDrone.test ├── VisDrone.train └── VisDrone.val ├── gen_dataset_UVADT.py ├── gen_dataset_visdrone.py ├── lib ├── __pycache__ │ ├── logger.cpython-38.pyc │ └── opts.cpython-38.pyc ├── cfg │ ├── UAVDT.json │ └── visdrone.json ├── datasets │ ├── __pycache__ │ │ └── dataset_factory.cpython-38.pyc │ ├── dataset │ │ ├── __pycache__ │ │ │ └── jde_stcmot.cpython-38.pyc │ │ └── jde_stcmot.py │ └── dataset_factory.py ├── logger.py ├── models │ ├── __pycache__ │ │ ├── data_parallel.cpython-38.pyc │ │ ├── decode.cpython-38.pyc │ │ ├── losses.cpython-38.pyc │ │ ├── model.cpython-38.pyc │ │ ├── scatter_gather.cpython-38.pyc │ │ └── utils.cpython-38.pyc │ ├── data_parallel.py │ ├── decode.py │ ├── losses.py │ ├── model.py │ ├── networks │ │ ├── STCMOT.py │ │ └── __pycache__ │ │ │ └── STCMOT.cpython-38.pyc │ ├── scatter_gather.py │ └── utils.py ├── opts.py ├── tracker │ ├── __pycache__ │ │ ├── basetrack.cpython-38.pyc │ │ ├── matching.cpython-38.pyc │ │ ├── multitracker.cpython-38.pyc │ │ └── multitracker_byte.cpython-38.pyc │ ├── basetrack.py │ ├── matching.py │ ├── multitracker.py │ └── multitracker_byte.py ├── tracking_utils │ ├── __pycache__ │ │ ├── evaluation.cpython-38.pyc │ │ ├── io.cpython-38.pyc │ │ ├── kalman_filter.cpython-38.pyc │ │ ├── log.cpython-38.pyc │ │ ├── timer.cpython-38.pyc │ │ ├── utils.cpython-38.pyc │ │ └── visualization.cpython-38.pyc │ ├── evaluation.py │ ├── io.py │ ├── kalman_filter.py │ ├── log.py │ ├── nms.py │ ├── parse_config.py │ ├── timer.py │ ├── utils.py │ └── visualization.py ├── trains │ ├── __pycache__ │ │ ├── base_trainer.cpython-38.pyc │ │ ├── mot.cpython-38.pyc │ │ └── train_factory.cpython-38.pyc │ ├── base_trainer.py │ ├── mot.py │ └── train_factory.py └── utils │ ├── __pycache__ │ ├── image.cpython-38.pyc │ ├── post_process.cpython-38.pyc │ └── utils.cpython-38.pyc │ ├── image.py │ ├── post_process.py │ └── utils.py ├── track.py └── train.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/README.md -------------------------------------------------------------------------------- /demo/pipeline.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/demo/pipeline.jpg -------------------------------------------------------------------------------- /experiments/train_stcmot.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/experiments/train_stcmot.sh -------------------------------------------------------------------------------- /models/ctdet_coco_dla_2x.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/models/ctdet_coco_dla_2x.pth -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/__pycache__/_init_paths.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/__pycache__/_init_paths.cpython-38.pyc -------------------------------------------------------------------------------- /src/__pycache__/gen_dataset_visdrone.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/__pycache__/gen_dataset_visdrone.cpython-38.pyc -------------------------------------------------------------------------------- /src/_init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/_init_paths.py -------------------------------------------------------------------------------- /src/data/UAVDT.test: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/data/UAVDT.test -------------------------------------------------------------------------------- /src/data/UAVDT.train: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/data/UAVDT.train -------------------------------------------------------------------------------- /src/data/VisDrone.test: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/data/VisDrone.test -------------------------------------------------------------------------------- /src/data/VisDrone.train: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/data/VisDrone.train -------------------------------------------------------------------------------- /src/data/VisDrone.val: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/data/VisDrone.val -------------------------------------------------------------------------------- /src/gen_dataset_UVADT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/gen_dataset_UVADT.py -------------------------------------------------------------------------------- /src/gen_dataset_visdrone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/gen_dataset_visdrone.py -------------------------------------------------------------------------------- /src/lib/__pycache__/logger.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/__pycache__/logger.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/__pycache__/opts.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/__pycache__/opts.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/cfg/UAVDT.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/cfg/UAVDT.json -------------------------------------------------------------------------------- /src/lib/cfg/visdrone.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/cfg/visdrone.json -------------------------------------------------------------------------------- /src/lib/datasets/__pycache__/dataset_factory.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/datasets/__pycache__/dataset_factory.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/datasets/dataset/__pycache__/jde_stcmot.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/datasets/dataset/__pycache__/jde_stcmot.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/datasets/dataset/jde_stcmot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/datasets/dataset/jde_stcmot.py -------------------------------------------------------------------------------- /src/lib/datasets/dataset_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/datasets/dataset_factory.py -------------------------------------------------------------------------------- /src/lib/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/logger.py -------------------------------------------------------------------------------- /src/lib/models/__pycache__/data_parallel.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/models/__pycache__/data_parallel.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/models/__pycache__/decode.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/models/__pycache__/decode.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/models/__pycache__/losses.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/models/__pycache__/losses.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/models/__pycache__/model.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/models/__pycache__/model.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/models/__pycache__/scatter_gather.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/models/__pycache__/scatter_gather.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/models/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/models/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/models/data_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/models/data_parallel.py -------------------------------------------------------------------------------- /src/lib/models/decode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/models/decode.py -------------------------------------------------------------------------------- /src/lib/models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/models/losses.py -------------------------------------------------------------------------------- /src/lib/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/models/model.py -------------------------------------------------------------------------------- /src/lib/models/networks/STCMOT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/models/networks/STCMOT.py -------------------------------------------------------------------------------- /src/lib/models/networks/__pycache__/STCMOT.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/models/networks/__pycache__/STCMOT.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/models/scatter_gather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/models/scatter_gather.py -------------------------------------------------------------------------------- /src/lib/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/models/utils.py -------------------------------------------------------------------------------- /src/lib/opts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/opts.py -------------------------------------------------------------------------------- /src/lib/tracker/__pycache__/basetrack.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracker/__pycache__/basetrack.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/tracker/__pycache__/matching.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracker/__pycache__/matching.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/tracker/__pycache__/multitracker.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracker/__pycache__/multitracker.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/tracker/__pycache__/multitracker_byte.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracker/__pycache__/multitracker_byte.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/tracker/basetrack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracker/basetrack.py -------------------------------------------------------------------------------- /src/lib/tracker/matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracker/matching.py -------------------------------------------------------------------------------- /src/lib/tracker/multitracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracker/multitracker.py -------------------------------------------------------------------------------- /src/lib/tracker/multitracker_byte.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracker/multitracker_byte.py -------------------------------------------------------------------------------- /src/lib/tracking_utils/__pycache__/evaluation.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/__pycache__/evaluation.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/tracking_utils/__pycache__/io.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/__pycache__/io.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/tracking_utils/__pycache__/kalman_filter.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/__pycache__/kalman_filter.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/tracking_utils/__pycache__/log.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/__pycache__/log.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/tracking_utils/__pycache__/timer.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/__pycache__/timer.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/tracking_utils/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/tracking_utils/__pycache__/visualization.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/__pycache__/visualization.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/tracking_utils/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/evaluation.py -------------------------------------------------------------------------------- /src/lib/tracking_utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/io.py -------------------------------------------------------------------------------- /src/lib/tracking_utils/kalman_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/kalman_filter.py -------------------------------------------------------------------------------- /src/lib/tracking_utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/log.py -------------------------------------------------------------------------------- /src/lib/tracking_utils/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/nms.py -------------------------------------------------------------------------------- /src/lib/tracking_utils/parse_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/parse_config.py -------------------------------------------------------------------------------- /src/lib/tracking_utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/timer.py -------------------------------------------------------------------------------- /src/lib/tracking_utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/utils.py -------------------------------------------------------------------------------- /src/lib/tracking_utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/tracking_utils/visualization.py -------------------------------------------------------------------------------- /src/lib/trains/__pycache__/base_trainer.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/trains/__pycache__/base_trainer.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/trains/__pycache__/mot.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/trains/__pycache__/mot.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/trains/__pycache__/train_factory.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/trains/__pycache__/train_factory.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/trains/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/trains/base_trainer.py -------------------------------------------------------------------------------- /src/lib/trains/mot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/trains/mot.py -------------------------------------------------------------------------------- /src/lib/trains/train_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/trains/train_factory.py -------------------------------------------------------------------------------- /src/lib/utils/__pycache__/image.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/utils/__pycache__/image.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/utils/__pycache__/post_process.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/utils/__pycache__/post_process.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/utils/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/utils/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /src/lib/utils/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/utils/image.py -------------------------------------------------------------------------------- /src/lib/utils/post_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/utils/post_process.py -------------------------------------------------------------------------------- /src/lib/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/lib/utils/utils.py -------------------------------------------------------------------------------- /src/track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/track.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ydhcg-BoBo/STCMOT/HEAD/src/train.py --------------------------------------------------------------------------------