├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── NOTICE ├── README.md ├── configs ├── .DS_Store ├── dataset │ └── cater.yaml └── experiment │ ├── .DS_Store │ ├── OC-MOT │ ├── _cater_bbox_mot_preprocessing.yaml │ ├── cater.yaml │ └── cater_eval.yaml │ ├── SAVi │ ├── _cater_bbox_mot_preprocessing.yaml │ └── cater.yaml │ └── _output_path.yaml ├── ocl ├── __init__.py ├── base.py ├── cli │ ├── cli_utils.py │ ├── compute_dataset_size.py │ ├── eval.py │ ├── eval_cluster_metrics.py │ ├── eval_utils.py │ └── train.py ├── combined_model.py ├── conditioning.py ├── config │ ├── __init__.py │ ├── conditioning.py │ ├── datasets.py │ ├── feature_extractors.py │ ├── memory.py │ ├── metrics.py │ ├── neural_networks.py │ ├── optimizers.py │ ├── perceptual_groupings.py │ ├── plugins.py │ ├── predictor.py │ └── utils.py ├── consistency.py ├── datasets.py ├── decoding.py ├── distillation.py ├── feature_extractors.py ├── hooks.py ├── losses.py ├── matching.py ├── memory.py ├── memory_rollout.py ├── metrics.py ├── mha.py ├── models │ ├── __init__.py │ ├── savi.py │ └── savi_with_memory.py ├── neural_networks.py ├── path_defaults.py ├── perceptual_grouping.py ├── plugins.py ├── predictor.py ├── preprocessing.py ├── scheduling.py ├── tree_utils.py ├── utils.py ├── visualization_types.py └── visualizations.py ├── poetry.lock ├── pyproject.toml ├── requirements.txt ├── setup.cfg └── srcs ├── cater_demo.gif ├── framework.png └── real_world_demo.gif /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/LICENSE -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- 1 | Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 2 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/README.md -------------------------------------------------------------------------------- /configs/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/configs/.DS_Store -------------------------------------------------------------------------------- /configs/dataset/cater.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/configs/dataset/cater.yaml -------------------------------------------------------------------------------- /configs/experiment/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/configs/experiment/.DS_Store -------------------------------------------------------------------------------- /configs/experiment/OC-MOT/_cater_bbox_mot_preprocessing.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/configs/experiment/OC-MOT/_cater_bbox_mot_preprocessing.yaml -------------------------------------------------------------------------------- /configs/experiment/OC-MOT/cater.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/configs/experiment/OC-MOT/cater.yaml -------------------------------------------------------------------------------- /configs/experiment/OC-MOT/cater_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/configs/experiment/OC-MOT/cater_eval.yaml -------------------------------------------------------------------------------- /configs/experiment/SAVi/_cater_bbox_mot_preprocessing.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/configs/experiment/SAVi/_cater_bbox_mot_preprocessing.yaml -------------------------------------------------------------------------------- /configs/experiment/SAVi/cater.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/configs/experiment/SAVi/cater.yaml -------------------------------------------------------------------------------- /configs/experiment/_output_path.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/configs/experiment/_output_path.yaml -------------------------------------------------------------------------------- /ocl/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ocl/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/base.py -------------------------------------------------------------------------------- /ocl/cli/cli_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/cli/cli_utils.py -------------------------------------------------------------------------------- /ocl/cli/compute_dataset_size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/cli/compute_dataset_size.py -------------------------------------------------------------------------------- /ocl/cli/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/cli/eval.py -------------------------------------------------------------------------------- /ocl/cli/eval_cluster_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/cli/eval_cluster_metrics.py -------------------------------------------------------------------------------- /ocl/cli/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/cli/eval_utils.py -------------------------------------------------------------------------------- /ocl/cli/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/cli/train.py -------------------------------------------------------------------------------- /ocl/combined_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/combined_model.py -------------------------------------------------------------------------------- /ocl/conditioning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/conditioning.py -------------------------------------------------------------------------------- /ocl/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/config/__init__.py -------------------------------------------------------------------------------- /ocl/config/conditioning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/config/conditioning.py -------------------------------------------------------------------------------- /ocl/config/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/config/datasets.py -------------------------------------------------------------------------------- /ocl/config/feature_extractors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/config/feature_extractors.py -------------------------------------------------------------------------------- /ocl/config/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/config/memory.py -------------------------------------------------------------------------------- /ocl/config/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/config/metrics.py -------------------------------------------------------------------------------- /ocl/config/neural_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/config/neural_networks.py -------------------------------------------------------------------------------- /ocl/config/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/config/optimizers.py -------------------------------------------------------------------------------- /ocl/config/perceptual_groupings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/config/perceptual_groupings.py -------------------------------------------------------------------------------- /ocl/config/plugins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/config/plugins.py -------------------------------------------------------------------------------- /ocl/config/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/config/predictor.py -------------------------------------------------------------------------------- /ocl/config/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/config/utils.py -------------------------------------------------------------------------------- /ocl/consistency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/consistency.py -------------------------------------------------------------------------------- /ocl/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/datasets.py -------------------------------------------------------------------------------- /ocl/decoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/decoding.py -------------------------------------------------------------------------------- /ocl/distillation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/distillation.py -------------------------------------------------------------------------------- /ocl/feature_extractors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/feature_extractors.py -------------------------------------------------------------------------------- /ocl/hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/hooks.py -------------------------------------------------------------------------------- /ocl/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/losses.py -------------------------------------------------------------------------------- /ocl/matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/matching.py -------------------------------------------------------------------------------- /ocl/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/memory.py -------------------------------------------------------------------------------- /ocl/memory_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/memory_rollout.py -------------------------------------------------------------------------------- /ocl/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/metrics.py -------------------------------------------------------------------------------- /ocl/mha.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/mha.py -------------------------------------------------------------------------------- /ocl/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/models/__init__.py -------------------------------------------------------------------------------- /ocl/models/savi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/models/savi.py -------------------------------------------------------------------------------- /ocl/models/savi_with_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/models/savi_with_memory.py -------------------------------------------------------------------------------- /ocl/neural_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/neural_networks.py -------------------------------------------------------------------------------- /ocl/path_defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/path_defaults.py -------------------------------------------------------------------------------- /ocl/perceptual_grouping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/perceptual_grouping.py -------------------------------------------------------------------------------- /ocl/plugins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/plugins.py -------------------------------------------------------------------------------- /ocl/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/predictor.py -------------------------------------------------------------------------------- /ocl/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/preprocessing.py -------------------------------------------------------------------------------- /ocl/scheduling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/scheduling.py -------------------------------------------------------------------------------- /ocl/tree_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/tree_utils.py -------------------------------------------------------------------------------- /ocl/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/utils.py -------------------------------------------------------------------------------- /ocl/visualization_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/visualization_types.py -------------------------------------------------------------------------------- /ocl/visualizations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/ocl/visualizations.py -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/setup.cfg -------------------------------------------------------------------------------- /srcs/cater_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/srcs/cater_demo.gif -------------------------------------------------------------------------------- /srcs/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/srcs/framework.png -------------------------------------------------------------------------------- /srcs/real_world_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/amazon-science/object-centric-multiple-object-tracking/HEAD/srcs/real_world_demo.gif --------------------------------------------------------------------------------