tracklab.wrappers.dataset.mot_like package

Submodules

tracklab.wrappers.dataset.mot_like.bee24 module

class tracklab.wrappers.dataset.mot_like.bee24.Bee24(dataset_path: str, nvid: int = -1, nframes: int = -1, vids_dict: list | None = None, public_dets_subpath: str | None = None, *args, **kwargs)

Bases: MOT

name = 'Bee24'

nickname = 'bee24'

tracklab.wrappers.dataset.mot_like.common module

class tracklab.wrappers.dataset.mot_like.common.MOT(dataset_path: str, categories_list: list, nvid: int = -1, nframes: int = -1, vids_dict: list | None = None, public_dets_subpath: str | None = None, leave_one_out_idx: int | None = None, *args, **kwargs)

Bases: TrackingDataset

load_set(dataset_path, nvid=-1, vids_filter_set=None)

load_set_wrapper(args)

process_trackeval_results(results, dataset_config, eval_config)

read_ini_file(file_path)

read_motchallenge_formatted_file(file_path)

read_motchallenge_result_formatted_file(file_path)

tracklab.wrappers.dataset.mot_like.dancetrack module

class tracklab.wrappers.dataset.mot_like.dancetrack.DanceTrack(dataset_path: str, nvid: int = -1, nframes: int = -1, vids_dict: list | None = None, public_dets_subpath: str | None = None, *args, **kwargs)

Bases: MOT

Public detections notes: - train:

• my_det.txt: official YoloX weights from DanceTrack, trained on train set, ran by Baptiste to get these detections

• val:

  • my_det.txt: official YoloX weights from DanceTrack, trained on train set, ran by Baptiste to get these detections

  • yolox_dets.txt: official YoloX detections from GHOST, trained on train set

• test

  • my_det.txt: official YoloX weights from DanceTrack, trained on train set, ran by Baptiste to get these detections

  • yolox_dets.txt: official YoloX detections from GHOST, trained on train (+val?) set

name = 'DanceTrack'

nickname = 'dt'

tracklab.wrappers.dataset.mot_like.mot17 module

class tracklab.wrappers.dataset.mot_like.mot17.MOT17(dataset_path: str, nvid: int = -1, nframes: int = -1, vids_dict: list | None = None, public_dets_subpath: str | None = None, *args, **kwargs)

Bases: MOT

name = 'MOT17'

nickname = 'm17'

tracklab.wrappers.dataset.mot_like.mot20 module

class tracklab.wrappers.dataset.mot_like.mot20.MOT20(dataset_path: str, nvid: int = -1, nframes: int = -1, vids_dict: list | None = None, public_dets_subpath: str | None = None, *args, **kwargs)

Bases: MOT

name = 'MOT20'

nickname = 'm20'

tracklab.wrappers.dataset.mot_like.sportsmot module

class tracklab.wrappers.dataset.mot_like.sportsmot.SportsMOT(dataset_path: str, nvid: int = -1, nframes: int = -1, vids_dict: list | None = None, public_dets_subpath: str | None = None, *args, **kwargs)

Bases: MOT

Public detections notes: - test

• dets.txt: official detections from deep_eiou

• diffmot_yolox_x.txt: official detections from diffmot, trained on train set

• diffmot_yolox_x_mix.txt: official detections from diffmot, trained on train + val set

• val

  • deep_eiou.txt: official YoloX weights from deep_eiou, ran by Baptiste to get hese detections. Train on train+val set

  • diffmot_yolox_x.txt: official detections from diffmot, trained on train set

  • diffmot_yolox_x_mix.txt: official detections from diffmot, trained on train + val set

• train

  • deep_eiou.txt: official YoloX weights from deep_eiou, ran by Baptiste to get hese detections. Train on train+val set

name = 'SportsMOT'

nickname = 'sm'