Skip to content

Commit 5d5474f

Browse files
committed
Update docs
1 parent f2dd487 commit 5d5474f

28 files changed

+1697
-12120
lines changed

README.md

+11-12
Original file line numberDiff line numberDiff line change
@@ -4,30 +4,29 @@
44
# Multi-object trackers in Python
55
Easy to use implementation of various multi-object tracking algorithms.
66

7-
[![DOI](https://zenodo.org/badge/148338463.svg)](https://zenodo.org/badge/latestdoi/148338463) [![Upload motrackers to PyPI](https://github.com/adipandas/multi-object-tracker/actions/workflows/python-publish.yml/badge.svg)](https://github.com/adipandas/multi-object-tracker/actions/workflows/python-publish.yml)
7+
[![DOI](https://zenodo.org/badge/148338463.svg)](https://zenodo.org/badge/latestdoi/148338463)
8+
[![Upload motrackers to PyPI](https://github.com/adipandas/multi-object-tracker/actions/workflows/python-publish.yml/badge.svg)](https://github.com/adipandas/multi-object-tracker/actions/workflows/python-publish.yml)
89

910

1011
`YOLOv3 + CentroidTracker` | `TF-MobileNetSSD + CentroidTracker`
1112
:-------------------------:|:-------------------------:
1213
![Cars with YOLO][cars-yolo-output] | ![Cows with tf-SSD][cows-tf-ssd-output]
1314
Video source: [link](https://flic.kr/p/L6qyxj) | Video source: [link](https://flic.kr/p/26WeEWy)
1415

16+
1517
## Available Multi Object Trackers
1618

17-
```
18-
CentroidTracker
19-
IOUTracker
20-
CentroidKF_Tracker
21-
SORT
22-
```
19+
- CentroidTracker
20+
- IOUTracker
21+
- CentroidKF_Tracker
22+
- SORT
23+
2324

2425
## Available OpenCV-based object detectors:
2526

26-
```
27-
detector.TF_SSDMobileNetV2
28-
detector.Caffe_SSDMobileNet
29-
detector.YOLOv3
30-
```
27+
- detector.TF_SSDMobileNetV2
28+
- detector.Caffe_SSDMobileNet
29+
- detector.YOLOv3
3130

3231
## Installation
3332

docs/.doctrees/environment.pickle

126 KB
Binary file not shown.
414 KB
Binary file not shown.
3.43 KB
Binary file not shown.

docs/.doctrees/index.doctree

23 Bytes
Binary file not shown.

docs/_modules/motrackers/centroid_kf_tracker.html

+2-2
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ <h1>Source code for motrackers.centroid_kf_tracker</h1><div class="highlight"><p
8787

8888

8989
<div class="viewcode-block" id="assign_tracks2detection_centroid_distances"><a class="viewcode-back" href="../../includeme/apidocuments.html#motrackers.centroid_kf_tracker.assign_tracks2detection_centroid_distances">[docs]</a><span class="k">def</span> <span class="nf">assign_tracks2detection_centroid_distances</span><span class="p">(</span><span class="n">bbox_tracks</span><span class="p">,</span> <span class="n">bbox_detections</span><span class="p">,</span> <span class="n">distance_threshold</span><span class="o">=</span><span class="mf">10.</span><span class="p">):</span>
90-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
90+
<span class="sd">&quot;&quot;&quot;</span>
9191
<span class="sd"> Assigns detected bounding boxes to tracked bounding boxes using IoU as a distance metric.</span>
9292

9393
<span class="sd"> Args:</span>
@@ -149,7 +149,7 @@ <h1>Source code for motrackers.centroid_kf_tracker</h1><div class="highlight"><p
149149

150150

151151
<div class="viewcode-block" id="CentroidKF_Tracker"><a class="viewcode-back" href="../../includeme/apidocuments.html#motrackers.centroid_kf_tracker.CentroidKF_Tracker">[docs]</a><span class="k">class</span> <span class="nc">CentroidKF_Tracker</span><span class="p">(</span><span class="n">Tracker</span><span class="p">):</span>
152-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
152+
<span class="sd">&quot;&quot;&quot;</span>
153153
<span class="sd"> Kalman filter based tracking of multiple detected objects.</span>
154154

155155
<span class="sd"> Args:</span>

docs/_modules/motrackers/detectors/caffe.html

+1-1
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ <h1>Source code for motrackers.detectors.caffe</h1><div class="highlight"><pre>
8383

8484

8585
<div class="viewcode-block" id="Caffe_SSDMobileNet"><a class="viewcode-back" href="../../../includeme/apidocuments.html#motrackers.detectors.caffe.Caffe_SSDMobileNet">[docs]</a><span class="k">class</span> <span class="nc">Caffe_SSDMobileNet</span><span class="p">(</span><span class="n">Detector</span><span class="p">):</span>
86-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
86+
<span class="sd">&quot;&quot;&quot;</span>
8787
<span class="sd"> Caffe SSD MobileNet model for Object Detection.</span>
8888

8989
<span class="sd"> Args:</span>

docs/_modules/motrackers/detectors/detector.html

+4-4
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ <h1>Source code for motrackers.detectors.detector</h1><div class="highlight"><pr
8383

8484

8585
<div class="viewcode-block" id="Detector"><a class="viewcode-back" href="../../../includeme/apidocuments.html#motrackers.detectors.detector.Detector">[docs]</a><span class="k">class</span> <span class="nc">Detector</span><span class="p">:</span>
86-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
86+
<span class="sd">&quot;&quot;&quot;</span>
8787
<span class="sd"> Abstract class for detector.</span>
8888

8989
<span class="sd"> Args:</span>
@@ -105,7 +105,7 @@ <h1>Source code for motrackers.detectors.detector</h1><div class="highlight"><pr
105105
<span class="bp">self</span><span class="o">.</span><span class="n">bbox_colors</span> <span class="o">=</span> <span class="p">{</span><span class="n">key</span><span class="p">:</span> <span class="n">np</span><span class="o">.</span><span class="n">random</span><span class="o">.</span><span class="n">randint</span><span class="p">(</span><span class="mi">0</span><span class="p">,</span> <span class="mi">255</span><span class="p">,</span> <span class="n">size</span><span class="o">=</span><span class="p">(</span><span class="mi">3</span><span class="p">,))</span><span class="o">.</span><span class="n">tolist</span><span class="p">()</span> <span class="k">for</span> <span class="n">key</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">object_names</span><span class="o">.</span><span class="n">keys</span><span class="p">()}</span>
106106

107107
<div class="viewcode-block" id="Detector.forward"><a class="viewcode-back" href="../../../includeme/apidocuments.html#motrackers.detectors.detector.Detector.forward">[docs]</a> <span class="k">def</span> <span class="nf">forward</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">image</span><span class="p">):</span>
108-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
108+
<span class="sd">&quot;&quot;&quot;</span>
109109
<span class="sd"> Forward pass for the detector with input image.</span>
110110

111111
<span class="sd"> Args:</span>
@@ -117,7 +117,7 @@ <h1>Source code for motrackers.detectors.detector</h1><div class="highlight"><pr
117117
<span class="k">raise</span> <span class="bp">NotImplemented</span></div>
118118

119119
<div class="viewcode-block" id="Detector.detect"><a class="viewcode-back" href="../../../includeme/apidocuments.html#motrackers.detectors.detector.Detector.detect">[docs]</a> <span class="k">def</span> <span class="nf">detect</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">image</span><span class="p">):</span>
120-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
120+
<span class="sd">&quot;&quot;&quot;</span>
121121
<span class="sd"> Detect objects in the input image.</span>
122122

123123
<span class="sd"> Args:</span>
@@ -157,7 +157,7 @@ <h1>Source code for motrackers.detectors.detector</h1><div class="highlight"><pr
157157
<span class="k">return</span> <span class="n">np</span><span class="o">.</span><span class="n">array</span><span class="p">([]),</span> <span class="n">np</span><span class="o">.</span><span class="n">array</span><span class="p">([]),</span> <span class="n">np</span><span class="o">.</span><span class="n">array</span><span class="p">([])</span></div>
158158

159159
<div class="viewcode-block" id="Detector.draw_bboxes"><a class="viewcode-back" href="../../../includeme/apidocuments.html#motrackers.detectors.detector.Detector.draw_bboxes">[docs]</a> <span class="k">def</span> <span class="nf">draw_bboxes</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">image</span><span class="p">,</span> <span class="n">bboxes</span><span class="p">,</span> <span class="n">confidences</span><span class="p">,</span> <span class="n">class_ids</span><span class="p">):</span>
160-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
160+
<span class="sd">&quot;&quot;&quot;</span>
161161
<span class="sd"> Draw the bounding boxes about detected objects in the image.</span>
162162

163163
<span class="sd"> Args:</span>

docs/_modules/motrackers/detectors/tf.html

+1-1
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ <h1>Source code for motrackers.detectors.tf</h1><div class="highlight"><pre>
8383

8484

8585
<div class="viewcode-block" id="TF_SSDMobileNetV2"><a class="viewcode-back" href="../../../includeme/apidocuments.html#motrackers.detectors.tf.TF_SSDMobileNetV2">[docs]</a><span class="k">class</span> <span class="nc">TF_SSDMobileNetV2</span><span class="p">(</span><span class="n">Detector</span><span class="p">):</span>
86-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
86+
<span class="sd">&quot;&quot;&quot;</span>
8787
<span class="sd"> Tensorflow SSD MobileNetv2 model for Object Detection.</span>
8888

8989
<span class="sd"> Args:</span>

docs/_modules/motrackers/detectors/yolo.html

+1-1
Original file line numberDiff line numberDiff line change
@@ -84,7 +84,7 @@ <h1>Source code for motrackers.detectors.yolo</h1><div class="highlight"><pre>
8484

8585

8686
<div class="viewcode-block" id="YOLOv3"><a class="viewcode-back" href="../../../includeme/apidocuments.html#motrackers.detectors.yolo.YOLOv3">[docs]</a><span class="k">class</span> <span class="nc">YOLOv3</span><span class="p">(</span><span class="n">Detector</span><span class="p">):</span>
87-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
87+
<span class="sd">&quot;&quot;&quot;</span>
8888
<span class="sd"> YOLOv3 Object Detector Module.</span>
8989

9090
<span class="sd"> Args:</span>

docs/_modules/motrackers/iou_tracker.html

+1-1
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ <h1>Source code for motrackers.iou_tracker</h1><div class="highlight"><pre>
8282

8383

8484
<div class="viewcode-block" id="IOUTracker"><a class="viewcode-back" href="../../includeme/apidocuments.html#motrackers.iou_tracker.IOUTracker">[docs]</a><span class="k">class</span> <span class="nc">IOUTracker</span><span class="p">(</span><span class="n">Tracker</span><span class="p">):</span>
85-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
85+
<span class="sd">&quot;&quot;&quot;</span>
8686
<span class="sd"> Intersection over Union Tracker.</span>
8787

8888
<span class="sd"> References</span>

docs/_modules/motrackers/kalman_tracker.html

+7-7
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ <h1>Source code for motrackers.kalman_tracker</h1><div class="highlight"><pre>
8181

8282

8383
<div class="viewcode-block" id="KalmanFilter"><a class="viewcode-back" href="../../includeme/apidocuments.html#motrackers.kalman_tracker.KalmanFilter">[docs]</a><span class="k">class</span> <span class="nc">KalmanFilter</span><span class="p">:</span>
84-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
84+
<span class="sd">&quot;&quot;&quot;</span>
8585
<span class="sd"> Kalman Filter Implementation.</span>
8686

8787
<span class="sd"> Args:</span>
@@ -124,7 +124,7 @@ <h1>Source code for motrackers.kalman_tracker</h1><div class="highlight"><pre>
124124
<span class="bp">self</span><span class="o">.</span><span class="n">x</span> <span class="o">=</span> <span class="n">np</span><span class="o">.</span><span class="n">zeros</span><span class="p">((</span><span class="bp">self</span><span class="o">.</span><span class="n">state_size</span><span class="p">,</span> <span class="mi">1</span><span class="p">))</span> <span class="k">if</span> <span class="n">initial_state</span> <span class="ow">is</span> <span class="kc">None</span> <span class="k">else</span> <span class="n">initial_state</span>
125125

126126
<div class="viewcode-block" id="KalmanFilter.predict"><a class="viewcode-back" href="../../includeme/apidocuments.html#motrackers.kalman_tracker.KalmanFilter.predict">[docs]</a> <span class="k">def</span> <span class="nf">predict</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">u</span><span class="o">=</span><span class="mi">0</span><span class="p">):</span>
127-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
127+
<span class="sd">&quot;&quot;&quot;</span>
128128
<span class="sd"> Prediction step of Kalman Filter.</span>
129129

130130
<span class="sd"> Args:</span>
@@ -143,7 +143,7 @@ <h1>Source code for motrackers.kalman_tracker</h1><div class="highlight"><pre>
143143
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">x</span></div>
144144

145145
<div class="viewcode-block" id="KalmanFilter.update"><a class="viewcode-back" href="../../includeme/apidocuments.html#motrackers.kalman_tracker.KalmanFilter.update">[docs]</a> <span class="k">def</span> <span class="nf">update</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">z</span><span class="p">):</span>
146-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
146+
<span class="sd">&quot;&quot;&quot;</span>
147147
<span class="sd"> Measurement update of Kalman Filter.</span>
148148

149149
<span class="sd"> Args:</span>
@@ -169,7 +169,7 @@ <h1>Source code for motrackers.kalman_tracker</h1><div class="highlight"><pre>
169169

170170

171171
<span class="k">def</span> <span class="nf">get_process_covariance_matrix</span><span class="p">(</span><span class="n">dt</span><span class="p">):</span>
172-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
172+
<span class="sd">&quot;&quot;&quot;</span>
173173
<span class="sd"> Generates a process noise covariance matrix for constant acceleration motion.</span>
174174

175175
<span class="sd"> Args:</span>
@@ -193,7 +193,7 @@ <h1>Source code for motrackers.kalman_tracker</h1><div class="highlight"><pre>
193193

194194

195195
<span class="k">def</span> <span class="nf">get_transition_matrix</span><span class="p">(</span><span class="n">dt</span><span class="p">):</span>
196-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
196+
<span class="sd">&quot;&quot;&quot;</span>
197197
<span class="sd"> Generate the transition matrix for constant acceleration motion.</span>
198198

199199
<span class="sd"> Args:</span>
@@ -207,7 +207,7 @@ <h1>Source code for motrackers.kalman_tracker</h1><div class="highlight"><pre>
207207

208208

209209
<div class="viewcode-block" id="KFTrackerConstantAcceleration"><a class="viewcode-back" href="../../includeme/apidocuments.html#motrackers.kalman_tracker.KFTrackerConstantAcceleration">[docs]</a><span class="k">class</span> <span class="nc">KFTrackerConstantAcceleration</span><span class="p">(</span><span class="n">KalmanFilter</span><span class="p">):</span>
210-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
210+
<span class="sd">&quot;&quot;&quot;</span>
211211
<span class="sd"> Kalman Filter with constant acceleration kinematic model.</span>
212212

213213
<span class="sd"> Args:</span>
@@ -279,7 +279,7 @@ <h1>Source code for motrackers.kalman_tracker</h1><div class="highlight"><pre>
279279

280280

281281
<div class="viewcode-block" id="KFTrackerSORT"><a class="viewcode-back" href="../../includeme/apidocuments.html#motrackers.kalman_tracker.KFTrackerSORT">[docs]</a><span class="k">class</span> <span class="nc">KFTrackerSORT</span><span class="p">(</span><span class="n">KalmanFilter</span><span class="p">):</span>
282-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
282+
<span class="sd">&quot;&quot;&quot;</span>
283283
<span class="sd"> Kalman filter for ``SORT``.</span>
284284

285285
<span class="sd"> Args:</span>

docs/_modules/motrackers/sort_tracker.html

+2-2
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ <h1>Source code for motrackers.sort_tracker</h1><div class="highlight"><pre>
8585

8686

8787
<div class="viewcode-block" id="assign_tracks2detection_iou"><a class="viewcode-back" href="../../includeme/apidocuments.html#motrackers.sort_tracker.assign_tracks2detection_iou">[docs]</a><span class="k">def</span> <span class="nf">assign_tracks2detection_iou</span><span class="p">(</span><span class="n">bbox_tracks</span><span class="p">,</span> <span class="n">bbox_detections</span><span class="p">,</span> <span class="n">iou_threshold</span><span class="o">=</span><span class="mf">0.3</span><span class="p">):</span>
88-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
88+
<span class="sd">&quot;&quot;&quot;</span>
8989
<span class="sd"> Assigns detected bounding boxes to tracked bounding boxes using IoU as a distance metric.</span>
9090

9191
<span class="sd"> Args:</span>
@@ -142,7 +142,7 @@ <h1>Source code for motrackers.sort_tracker</h1><div class="highlight"><pre>
142142

143143

144144
<div class="viewcode-block" id="SORT"><a class="viewcode-back" href="../../includeme/apidocuments.html#motrackers.sort_tracker.SORT">[docs]</a><span class="k">class</span> <span class="nc">SORT</span><span class="p">(</span><span class="n">CentroidKF_Tracker</span><span class="p">):</span>
145-
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
145+
<span class="sd">&quot;&quot;&quot;</span>
146146
<span class="sd"> SORT - Multi object tracker.</span>
147147

148148
<span class="sd"> Args:</span>

0 commit comments

Comments
 (0)