OfflineMOT
is a package for multi objects tracking from bird's eye view stationary videos. The accuracy has priority over runtime in this package, therefore it is better suited for offline processing rather than real time applications, hence the name of the package.
Update 21/2/2023: New changes are made to the tracking, like using GOTURN and the option to use kalman filter. Sadly, the documentation is not updated yet, but you can control the new configuration from the configuration file.
Update 27/3/2023: The ability for manaul intervention by pressing s
key to save the current tracking state is added. This is useful when the tracking is not working well and you want to correct it manually. The documentation is not yet updated.
A pretrained Yolo network is used for detection in this package and it should be trained separately. The network included with this library is Yolo v4 in a pytorch format (trained to detect pedestrians, cyclists and cars). The loading and running of Yolo model is done with the help of scripts taken from this project (All of them are in offlinemot/tool subfolder)
For training YOLOv4 from scratch the same repo is useful. Additionally, Darknet code can be used to train YOLO and then converting the resulted trained file to pytorch format .pth
as described by the docs.
Example output for a sample video, taken from Vehicle-Crowd Interaction (VCI) - CITR Dataset :
This example shows some minor problems because the scene, the background and the drone's camera are outside the detection network training set (never seen before by the detection network).
However, the application of this project (including the Yolo network training) was targeted for a Cyclists dataset videos [to be cited later].
The package can be installed on python 3.x simply using the pip
command:
pip install offlinemot
For developers
To work on the latest development branch, one can simple clone the repo
git clone https://github.com/engyasin/Offline_MOT
and then from the repo root run:
pip install -e .
If you use this software in your work, please cite it as following
@article{Yousif2022,
doi = {10.21105/joss.04099},
url = {https://doi.org/10.21105/joss.04099},
year = {2022},
publisher = {The Open Journal},
volume = {7},
number = {74},
pages = {4099},
author = {Yasin Maan Yousif and Awad Mukbil and Jörg P. Müller},
title = {OfflineMOT: A Python package for multiple objects detection and tracking from bird view stationary drone videos},
journal = {Journal of Open Source Software}
}
The documentation includes some example and guides to run this package and it is available here https://engyasin.github.io/Offline_MOT/
Jupyter notebooks tutorials format are also available here
A technical report can be found here
After installing the library, and in order to test the example provided with it, the following lines can be used in as python commands:
In [1]: import offlinemot
In [2]: from offlinemot.config import configs
In [3]: cfg = configs() # if you have avaliable configuration file '.ini', you can pass it
In [4]: cfg.print_summary() # show the current values and sections
In [5]: cfg['detect_every_N'] = 3
In [6]: cfg.print_section('Detection') # show parameters of single section
In [7]: cfg['detect_thresh'] = 15
In [8]: offlinemot.core.extract_paths(config=cfg) # no input to run the example video
In [9]: cfg.write('new_config_file.ini') # to be loaded for similar videos
Lines 2 through 7 import and change the parameters for running the program.
For the first time line 8 is ran, the example network model will be downloaded (around 250MB). And a window for the example video with the tracked objects will be shown.
Line 9 save the current set of parameters into the provided file name.
The tracked objects will be surrounded with boxes in 5 different colors. Each color has a spicific meaning:
- Green: Pedestrian is detected.
- Blue: Cyclist is detected.
- Black: Car is detected.
- Red: The tracked object has failed the tracking step for the current frame
- White: The object is moving but still not classified to a class.
Of course, for a different case, the colors can be changed from the configs
class attribute (colors_map
). This also depends on the number of classes to predict.
To control this parameter and many others, the values can be assigned to the configs
instance:
cfg = configs() #input can be named file, or empty for the default values.
cfg['colors_map'] = [(255,0,0),(0,0,255),(0,255,0)]
Note: It's highly recommended to set all the parameters when running on a new video. A detailed description for their meaning is available in the config
file. Additionally, a complete example for parameters tuning is available in the documentation here
Then to run it on a new video, the command is:
offlinemot.core.extract_paths('path_to_video',config=cfg)
#[directory of the videos, leave empty to run the example video]
to show the result on the same video after post processing, use the command:
offlinemot.show_results.show_result('path_to_same_video',config=cfg)
#[directory of the videos, leave empty to run the example video]
Finally, to change the yolo network used in the package, the complete directory to 3 files need to be assigned through configs
class:
- .pth for the model weights
cfg['model_name'] = 'directory'
- .cfg for the Yolo configuration.
cfg['model_config'] = 'directory'
- .names for a simple text file containing the names of the classes.
cfg['class_file_name'] = 'directory'
This project can be used for:
-
Traffic trajectories extraction from videos (It is originally built to extract trajectories for a cyclist's dataset for traffic modelling research recorded in TU-Clausthal).
-
Tracking other objects (like animals) from bird's eye view in an offline manner.
There are a number of test units for this project. If a development of the package is intended then they can be run after cloning this repo with the command:
$ pytest -v ./offlinemot/tests
For the previous command pytest
library is needed to be installed.
Additionally, upon pushes and pull requests a github action would run pytest as well.
If you like to contribute to a feature of a bug fix, please take a look at the contribution instructions page. It has further details.
Alternatively, you can contribute by creating an issue for a problem when running the program. If your issue is about the accuracy of the results (like not detecting or failing to track some objects), please tag the issue with logic error. Please also attach some images or gif files depicting how the error happened in running and post-running time of the video.
Please star this repository if you find it useful, or use it as part of your research.
OfflineMOT
is free software and is licensed under the MIT License. Copyright (c) 2022, Yasin Yousif