added CSV output of Tracking data
This commit is contained in:
+121
@@ -0,0 +1,121 @@
|
||||
Distance
|
||||
[0. 0.]
|
||||
[ 9.1552734e-05 -2.0001221e-01]
|
||||
[-0.00082397 -0.47198486]
|
||||
[-1.1444092e-03 -1.2733459e+00]
|
||||
[-0.00241089 -2.2083435 ]
|
||||
[-0.00376892 -3.3737793 ]
|
||||
[-1.3732910e-03 -4.7552795e+00]
|
||||
[-2.1667480e-03 -6.4179993e+00]
|
||||
[-1.4190674e-03 -8.3665466e+00]
|
||||
[-3.81469727e-03 -1.04045105e+01]
|
||||
[-4.1961670e-03 -1.2636627e+01]
|
||||
[-1.7852783e-03 -1.5272430e+01]
|
||||
[-2.0751953e-03 -1.7859528e+01]
|
||||
[-2.9449463e-03 -2.0609467e+01]
|
||||
[-2.7313232e-03 -2.3581024e+01]
|
||||
[-3.1127930e-03 -2.6805878e+01]
|
||||
[-2.4719238e-03 -3.0153046e+01]
|
||||
[-8.544922e-04 -3.345955e+01]
|
||||
[-2.1972656e-03 -3.7096069e+01]
|
||||
[-3.479004e-03 -4.063263e+01]
|
||||
[-1.0528564e-03 -4.4414154e+01]
|
||||
[-3.5552979e-03 -4.8352509e+01]
|
||||
[-1.9378662e-03 -5.2272339e+01]
|
||||
[-3.1585693e-03 -5.6271851e+01]
|
||||
[-4.8828125e-03 -6.0289413e+01]
|
||||
[-6.1035156e-05 -6.4382584e+01]
|
||||
[-2.4414062e-03 -6.8487839e+01]
|
||||
[-1.3275146e-03 -7.2667847e+01]
|
||||
[-3.8299561e-03 -7.6995605e+01]
|
||||
[-2.1972656e-03 -8.1272278e+01]
|
||||
[-2.2125244e-03 -8.5408813e+01]
|
||||
[-4.4250488e-03 -8.9561066e+01]
|
||||
[-3.2348633e-03 -9.3909775e+01]
|
||||
[-1.2664795e-03 -9.8144012e+01]
|
||||
[-1.6937256e-03 -1.0228096e+02]
|
||||
[-2.1667480e-03 -1.0636488e+02]
|
||||
[-3.1433105e-03 -1.1039737e+02]
|
||||
[-1.4648438e-03 -1.1439691e+02]
|
||||
[-2.5482178e-03 -1.1836572e+02]
|
||||
[-4.2877197e-03 -1.2221335e+02]
|
||||
[-3.6468506e-03 -1.2595743e+02]
|
||||
[-1.5869141e-03 -1.2948502e+02]
|
||||
[-9.613037e-04 -1.331497e+02]
|
||||
[-9.1552734e-04 -1.3646295e+02]
|
||||
[-2.5634766e-03 -1.3976447e+02]
|
||||
[-1.3122559e-03 -1.4295718e+02]
|
||||
[-2.3498535e-03 -1.4595740e+02]
|
||||
[-3.0364990e-03 -1.4870921e+02]
|
||||
[-1.9226074e-03 -1.5140445e+02]
|
||||
[-4.0893555e-03 -1.5395099e+02]
|
||||
[-2.7313232e-03 -1.5627090e+02]
|
||||
[-2.8381348e-03 -1.5829169e+02]
|
||||
[-2.2125244e-03 -1.6021062e+02]
|
||||
[-1.8157959e-03 -1.6180893e+02]
|
||||
[-2.7160645e-03 -1.6327850e+02]
|
||||
[-1.4190674e-03 -1.6441026e+02]
|
||||
[-2.6550293e-03 -1.6538528e+02]
|
||||
[-1.663208e-03 -1.661462e+02]
|
||||
[-2.5634766e-03 -1.6644896e+02]
|
||||
[-2.0141602e-03 -1.6653844e+02]
|
||||
[-2.5634766e-03 -1.6644896e+02]
|
||||
[-1.4801025e-03 -1.6614972e+02]
|
||||
[-2.6550293e-03 -1.6540744e+02]
|
||||
[-1.4343262e-03 -1.6445952e+02]
|
||||
[-1.6937256e-03 -1.6337158e+02]
|
||||
[-8.3923340e-04 -1.6199916e+02]
|
||||
[-1.9836426e-03 -1.6037265e+02]
|
||||
[-2.9449463e-03 -1.5848950e+02]
|
||||
[-2.5482178e-03 -1.5645944e+02]
|
||||
[-4.4403076e-03 -1.5435570e+02]
|
||||
[-2.5634766e-03 -1.5195375e+02]
|
||||
[-1.6784668e-03 -1.4936133e+02]
|
||||
[-3.3264160e-03 -1.4648756e+02]
|
||||
[-2.3345947e-03 -1.4355959e+02]
|
||||
[-3.9520264e-03 -1.4051952e+02]
|
||||
[-2.6702881e-03 -1.3739568e+02]
|
||||
[-1.6479492e-03 -1.3414554e+02]
|
||||
[-2.9144287e-03 -1.3053851e+02]
|
||||
[-2.74658203e-04 -1.27146194e+02]
|
||||
[-4.6386719e-03 -1.2341406e+02]
|
||||
[-3.82995605e-03 -1.19604935e+02]
|
||||
[-1.6784668e-03 -1.1585806e+02]
|
||||
[-2.7465820e-03 -1.1199983e+02]
|
||||
[-1.4953613e-03 -1.0800525e+02]
|
||||
[-2.8839111e-03 -1.0399945e+02]
|
||||
[-2.029419e-03 -9.990352e+01]
|
||||
[-3.2348633e-03 -9.5714462e+01]
|
||||
[-2.822876e-03 -9.153703e+01]
|
||||
[-2.0599365e-03 -8.7430099e+01]
|
||||
[-2.8686523e-03 -8.3358780e+01]
|
||||
[-1.8005371e-03 -7.9208038e+01]
|
||||
[-3.3569336e-03 -7.5048431e+01]
|
||||
[-1.9073486e-03 -7.0858276e+01]
|
||||
[-3.4484863e-03 -6.6669250e+01]
|
||||
[-1.7547607e-03 -6.2581085e+01]
|
||||
[-3.6468506e-03 -5.8541046e+01]
|
||||
[-4.1656494e-03 -5.4584732e+01]
|
||||
[-1.4953613e-03 -5.0713928e+01]
|
||||
[-3.4637451e-03 -4.6957169e+01]
|
||||
[-2.6550293e-03 -4.3271194e+01]
|
||||
[-2.1820068e-03 -3.9475052e+01]
|
||||
[-1.8310547e-03 -3.5997910e+01]
|
||||
[-1.5106201e-03 -3.2473053e+01]
|
||||
[-2.2125244e-03 -2.9272049e+01]
|
||||
[-3.9825439e-03 -2.6096252e+01]
|
||||
[-8.6975098e-04 -2.2999756e+01]
|
||||
[-2.5939941e-03 -2.0100464e+01]
|
||||
[-2.0751953e-03 -1.7358124e+01]
|
||||
[-2.6397705e-03 -1.4636017e+01]
|
||||
[-3.6621094e-03 -1.2362274e+01]
|
||||
[-3.7689209e-03 -1.0150055e+01]
|
||||
[-3.5095215e-04 -8.1454773e+00]
|
||||
[-8.8500977e-04 -6.2778015e+00]
|
||||
[-1.9073486e-03 -4.5388184e+00]
|
||||
[-0.00395203 -3.2828064 ]
|
||||
[-1.7700195e-03 -2.1487122e+00]
|
||||
[-1.1291504e-03 -1.2661133e+00]
|
||||
[-0.00056458 -0.46047974]
|
||||
[ 9.1552734e-05 -2.0001221e-01]
|
||||
[0. 0.]
|
||||
|
+121
@@ -0,0 +1,121 @@
|
||||
Distance
|
||||
[0. 0.]
|
||||
[-3.0517578e-04 -3.6169434e-01]
|
||||
[ 6.1035156e-04 -1.3816223e+00]
|
||||
[-1.7700195e-03 -3.1427612e+00]
|
||||
[-3.6621094e-04 -5.3888245e+00]
|
||||
[ 8.544922e-04 -8.352051e+00]
|
||||
[-1.6784668e-03 -1.1628967e+01]
|
||||
[ 4.5776367e-04 -1.5514008e+01]
|
||||
[-3.9672852e-04 -2.0042877e+01]
|
||||
[-1.4343262e-03 -2.4713379e+01]
|
||||
[-1.6479492e-03 -2.9954529e+01]
|
||||
[-9.1552734e-05 -3.5395569e+01]
|
||||
[ 6.1035156e-05 -4.1268753e+01]
|
||||
[ 1.6784668e-03 -4.7279175e+01]
|
||||
[ 6.1035156e-05 -5.3451248e+01]
|
||||
[-6.4086914e-04 -5.9953857e+01]
|
||||
[-7.019043e-04 -6.645683e+01]
|
||||
[-4.5776367e-04 -7.3268570e+01]
|
||||
[-9.460449e-04 -7.995618e+01]
|
||||
[-1.4038086e-03 -8.6606598e+01]
|
||||
[-1.1901855e-03 -9.3402725e+01]
|
||||
[-9.76562500e-04 -1.00152145e+02]
|
||||
[-1.0681152e-03 -1.0658214e+02]
|
||||
[-1.9836426e-03 -1.1319769e+02]
|
||||
[-1.19018555e-03 -1.19383316e+02]
|
||||
[-1.4343262e-03 -1.2540285e+02]
|
||||
[-1.9836426e-03 -1.3126787e+02]
|
||||
[-1.2207031e-04 -1.3664021e+02]
|
||||
[-8.5449219e-04 -1.4185736e+02]
|
||||
[-1.1291504e-03 -1.4654036e+02]
|
||||
[-2.6245117e-03 -1.5104810e+02]
|
||||
[-1.2207031e-04 -1.5495306e+02]
|
||||
[-1.6479492e-03 -1.5836139e+02]
|
||||
[-2.1362305e-04 -1.6127242e+02]
|
||||
[ 0.7775574 -158.7493 ]
|
||||
[-3.34777832e-02 -1.01322174e+02]
|
||||
[-3.4423828e-02 -1.0185884e+02]
|
||||
[-3.6437988e-02 -1.0195259e+02]
|
||||
[-3.2470703e-02 -1.0182524e+02]
|
||||
[-3.1555176e-02 -1.0128633e+02]
|
||||
[ 0.7785034 -158.69778 ]
|
||||
[-6.1035156e-05 -1.6095703e+02]
|
||||
[-1.8920898e-03 -1.5795256e+02]
|
||||
[-1.7089844e-03 -1.5435452e+02]
|
||||
[-1.0986328e-03 -1.5027046e+02]
|
||||
[-1.7089844e-03 -1.4552046e+02]
|
||||
[-1.9531250e-03 -1.4051744e+02]
|
||||
[ 1.6479492e-03 -1.3527052e+02]
|
||||
[ 2.4414062e-04 -1.2941704e+02]
|
||||
[-3.05175781e-04 -1.23411545e+02]
|
||||
[ 3.9672852e-04 -1.1720087e+02]
|
||||
[ 2.13623047e-03 -1.10537766e+02]
|
||||
[ 6.1035156e-04 -1.0399693e+02]
|
||||
[ 3.3569336e-04 -9.7202988e+01]
|
||||
[-1.739502e-03 -9.027710e+01]
|
||||
[-4.272461e-04 -8.335503e+01]
|
||||
[-1.1901855e-03 -7.6376801e+01]
|
||||
[-4.8828125e-04 -6.9428574e+01]
|
||||
[-7.6293945e-04 -6.2581680e+01]
|
||||
[-8.8500977e-04 -5.6045334e+01]
|
||||
[-1.5869141e-03 -4.9427963e+01]
|
||||
[-9.7656250e-04 -4.3270126e+01]
|
||||
[-1.5869141e-03 -3.7204178e+01]
|
||||
[-1.4343262e-03 -3.1393860e+01]
|
||||
[-1.7395020e-03 -2.6090973e+01]
|
||||
[-5.798340e-04 -2.104535e+01]
|
||||
[-8.5449219e-04 -1.6401184e+01]
|
||||
[-1.0070801e-03 -1.2360779e+01]
|
||||
[ 7.9345703e-04 -8.6397400e+00]
|
||||
[ 7.324219e-04 -5.603607e+00]
|
||||
[-2.5939941e-03 -3.2775574e+00]
|
||||
[ 2.4414062e-04 -1.4170227e+00]
|
||||
[ 3.6621094e-04 -3.7210083e-01]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
[0. 0.]
|
||||
|
@@ -2,6 +2,7 @@ import cv2
|
||||
import numpy as np
|
||||
import argparse
|
||||
import sys
|
||||
import csv
|
||||
|
||||
from util import image_crop, to_rect, bbox_add_position, bbox_center, bbox_round
|
||||
|
||||
@@ -301,6 +302,7 @@ if __name__ == '__main__':
|
||||
parser.add_argument("--track", action="store_true")
|
||||
parser.add_argument("--path", action="store_true")
|
||||
parser.add_argument("--loop", action="store_true")
|
||||
parser.add_argument("--name", default="Default")
|
||||
args = parser.parse_args()
|
||||
video = cv2.VideoCapture(args.filename)
|
||||
|
||||
@@ -337,6 +339,7 @@ if __name__ == '__main__':
|
||||
key_wait = -1
|
||||
dist_min = [(0, 0)]*len(tracker_list)
|
||||
dist_max = [(0, 0)]*len(tracker_list)
|
||||
result_data = [[] for i in range(len(tracker_list))]
|
||||
while True:
|
||||
# Start timer
|
||||
timer = cv2.getTickCount()
|
||||
@@ -355,6 +358,7 @@ if __name__ == '__main__':
|
||||
for ct in tracker_list:
|
||||
mean_distance = ct.process(image, image_anno)
|
||||
if mean_distance is not None:
|
||||
result_data[i].append({f'Distance': mean_distance})
|
||||
scale = ct_params.scale
|
||||
scaled_distance = (scale*mean_distance[0], scale*mean_distance[1])
|
||||
dist_min[i] = (min(scaled_distance[0], dist_min[i][0]), min(scaled_distance[1], dist_min[i][1]))
|
||||
@@ -383,3 +387,16 @@ if __name__ == '__main__':
|
||||
else:
|
||||
key_wait = -1
|
||||
|
||||
# writing to csv file
|
||||
i = 1
|
||||
for result in result_data:
|
||||
with open(f"{args.name}_{i}.csv", 'w') as csvfile:
|
||||
# creating a csv dict writer object
|
||||
writer = csv.DictWriter(csvfile, fieldnames=['Distance'])
|
||||
|
||||
# writing headers (field names)
|
||||
writer.writeheader()
|
||||
|
||||
# writing data rows
|
||||
writer.writerows(result)
|
||||
i += 1
|
||||
|
||||
Reference in New Issue
Block a user