added CSV output of Tracking data

This commit is contained in:
2024-07-09 08:59:03 +02:00
parent 1a879c2820
commit a6bb1bf3f8
3 changed files with 259 additions and 0 deletions
+121
View File
@@ -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.]
1 Distance
2 [0. 0.]
3 [ 9.1552734e-05 -2.0001221e-01]
4 [-0.00082397 -0.47198486]
5 [-1.1444092e-03 -1.2733459e+00]
6 [-0.00241089 -2.2083435 ]
7 [-0.00376892 -3.3737793 ]
8 [-1.3732910e-03 -4.7552795e+00]
9 [-2.1667480e-03 -6.4179993e+00]
10 [-1.4190674e-03 -8.3665466e+00]
11 [-3.81469727e-03 -1.04045105e+01]
12 [-4.1961670e-03 -1.2636627e+01]
13 [-1.7852783e-03 -1.5272430e+01]
14 [-2.0751953e-03 -1.7859528e+01]
15 [-2.9449463e-03 -2.0609467e+01]
16 [-2.7313232e-03 -2.3581024e+01]
17 [-3.1127930e-03 -2.6805878e+01]
18 [-2.4719238e-03 -3.0153046e+01]
19 [-8.544922e-04 -3.345955e+01]
20 [-2.1972656e-03 -3.7096069e+01]
21 [-3.479004e-03 -4.063263e+01]
22 [-1.0528564e-03 -4.4414154e+01]
23 [-3.5552979e-03 -4.8352509e+01]
24 [-1.9378662e-03 -5.2272339e+01]
25 [-3.1585693e-03 -5.6271851e+01]
26 [-4.8828125e-03 -6.0289413e+01]
27 [-6.1035156e-05 -6.4382584e+01]
28 [-2.4414062e-03 -6.8487839e+01]
29 [-1.3275146e-03 -7.2667847e+01]
30 [-3.8299561e-03 -7.6995605e+01]
31 [-2.1972656e-03 -8.1272278e+01]
32 [-2.2125244e-03 -8.5408813e+01]
33 [-4.4250488e-03 -8.9561066e+01]
34 [-3.2348633e-03 -9.3909775e+01]
35 [-1.2664795e-03 -9.8144012e+01]
36 [-1.6937256e-03 -1.0228096e+02]
37 [-2.1667480e-03 -1.0636488e+02]
38 [-3.1433105e-03 -1.1039737e+02]
39 [-1.4648438e-03 -1.1439691e+02]
40 [-2.5482178e-03 -1.1836572e+02]
41 [-4.2877197e-03 -1.2221335e+02]
42 [-3.6468506e-03 -1.2595743e+02]
43 [-1.5869141e-03 -1.2948502e+02]
44 [-9.613037e-04 -1.331497e+02]
45 [-9.1552734e-04 -1.3646295e+02]
46 [-2.5634766e-03 -1.3976447e+02]
47 [-1.3122559e-03 -1.4295718e+02]
48 [-2.3498535e-03 -1.4595740e+02]
49 [-3.0364990e-03 -1.4870921e+02]
50 [-1.9226074e-03 -1.5140445e+02]
51 [-4.0893555e-03 -1.5395099e+02]
52 [-2.7313232e-03 -1.5627090e+02]
53 [-2.8381348e-03 -1.5829169e+02]
54 [-2.2125244e-03 -1.6021062e+02]
55 [-1.8157959e-03 -1.6180893e+02]
56 [-2.7160645e-03 -1.6327850e+02]
57 [-1.4190674e-03 -1.6441026e+02]
58 [-2.6550293e-03 -1.6538528e+02]
59 [-1.663208e-03 -1.661462e+02]
60 [-2.5634766e-03 -1.6644896e+02]
61 [-2.0141602e-03 -1.6653844e+02]
62 [-2.5634766e-03 -1.6644896e+02]
63 [-1.4801025e-03 -1.6614972e+02]
64 [-2.6550293e-03 -1.6540744e+02]
65 [-1.4343262e-03 -1.6445952e+02]
66 [-1.6937256e-03 -1.6337158e+02]
67 [-8.3923340e-04 -1.6199916e+02]
68 [-1.9836426e-03 -1.6037265e+02]
69 [-2.9449463e-03 -1.5848950e+02]
70 [-2.5482178e-03 -1.5645944e+02]
71 [-4.4403076e-03 -1.5435570e+02]
72 [-2.5634766e-03 -1.5195375e+02]
73 [-1.6784668e-03 -1.4936133e+02]
74 [-3.3264160e-03 -1.4648756e+02]
75 [-2.3345947e-03 -1.4355959e+02]
76 [-3.9520264e-03 -1.4051952e+02]
77 [-2.6702881e-03 -1.3739568e+02]
78 [-1.6479492e-03 -1.3414554e+02]
79 [-2.9144287e-03 -1.3053851e+02]
80 [-2.74658203e-04 -1.27146194e+02]
81 [-4.6386719e-03 -1.2341406e+02]
82 [-3.82995605e-03 -1.19604935e+02]
83 [-1.6784668e-03 -1.1585806e+02]
84 [-2.7465820e-03 -1.1199983e+02]
85 [-1.4953613e-03 -1.0800525e+02]
86 [-2.8839111e-03 -1.0399945e+02]
87 [-2.029419e-03 -9.990352e+01]
88 [-3.2348633e-03 -9.5714462e+01]
89 [-2.822876e-03 -9.153703e+01]
90 [-2.0599365e-03 -8.7430099e+01]
91 [-2.8686523e-03 -8.3358780e+01]
92 [-1.8005371e-03 -7.9208038e+01]
93 [-3.3569336e-03 -7.5048431e+01]
94 [-1.9073486e-03 -7.0858276e+01]
95 [-3.4484863e-03 -6.6669250e+01]
96 [-1.7547607e-03 -6.2581085e+01]
97 [-3.6468506e-03 -5.8541046e+01]
98 [-4.1656494e-03 -5.4584732e+01]
99 [-1.4953613e-03 -5.0713928e+01]
100 [-3.4637451e-03 -4.6957169e+01]
101 [-2.6550293e-03 -4.3271194e+01]
102 [-2.1820068e-03 -3.9475052e+01]
103 [-1.8310547e-03 -3.5997910e+01]
104 [-1.5106201e-03 -3.2473053e+01]
105 [-2.2125244e-03 -2.9272049e+01]
106 [-3.9825439e-03 -2.6096252e+01]
107 [-8.6975098e-04 -2.2999756e+01]
108 [-2.5939941e-03 -2.0100464e+01]
109 [-2.0751953e-03 -1.7358124e+01]
110 [-2.6397705e-03 -1.4636017e+01]
111 [-3.6621094e-03 -1.2362274e+01]
112 [-3.7689209e-03 -1.0150055e+01]
113 [-3.5095215e-04 -8.1454773e+00]
114 [-8.8500977e-04 -6.2778015e+00]
115 [-1.9073486e-03 -4.5388184e+00]
116 [-0.00395203 -3.2828064 ]
117 [-1.7700195e-03 -2.1487122e+00]
118 [-1.1291504e-03 -1.2661133e+00]
119 [-0.00056458 -0.46047974]
120 [ 9.1552734e-05 -2.0001221e-01]
121 [0. 0.]
+121
View File
@@ -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.]
1 Distance
2 [0. 0.]
3 [-3.0517578e-04 -3.6169434e-01]
4 [ 6.1035156e-04 -1.3816223e+00]
5 [-1.7700195e-03 -3.1427612e+00]
6 [-3.6621094e-04 -5.3888245e+00]
7 [ 8.544922e-04 -8.352051e+00]
8 [-1.6784668e-03 -1.1628967e+01]
9 [ 4.5776367e-04 -1.5514008e+01]
10 [-3.9672852e-04 -2.0042877e+01]
11 [-1.4343262e-03 -2.4713379e+01]
12 [-1.6479492e-03 -2.9954529e+01]
13 [-9.1552734e-05 -3.5395569e+01]
14 [ 6.1035156e-05 -4.1268753e+01]
15 [ 1.6784668e-03 -4.7279175e+01]
16 [ 6.1035156e-05 -5.3451248e+01]
17 [-6.4086914e-04 -5.9953857e+01]
18 [-7.019043e-04 -6.645683e+01]
19 [-4.5776367e-04 -7.3268570e+01]
20 [-9.460449e-04 -7.995618e+01]
21 [-1.4038086e-03 -8.6606598e+01]
22 [-1.1901855e-03 -9.3402725e+01]
23 [-9.76562500e-04 -1.00152145e+02]
24 [-1.0681152e-03 -1.0658214e+02]
25 [-1.9836426e-03 -1.1319769e+02]
26 [-1.19018555e-03 -1.19383316e+02]
27 [-1.4343262e-03 -1.2540285e+02]
28 [-1.9836426e-03 -1.3126787e+02]
29 [-1.2207031e-04 -1.3664021e+02]
30 [-8.5449219e-04 -1.4185736e+02]
31 [-1.1291504e-03 -1.4654036e+02]
32 [-2.6245117e-03 -1.5104810e+02]
33 [-1.2207031e-04 -1.5495306e+02]
34 [-1.6479492e-03 -1.5836139e+02]
35 [-2.1362305e-04 -1.6127242e+02]
36 [ 0.7775574 -158.7493 ]
37 [-3.34777832e-02 -1.01322174e+02]
38 [-3.4423828e-02 -1.0185884e+02]
39 [-3.6437988e-02 -1.0195259e+02]
40 [-3.2470703e-02 -1.0182524e+02]
41 [-3.1555176e-02 -1.0128633e+02]
42 [ 0.7785034 -158.69778 ]
43 [-6.1035156e-05 -1.6095703e+02]
44 [-1.8920898e-03 -1.5795256e+02]
45 [-1.7089844e-03 -1.5435452e+02]
46 [-1.0986328e-03 -1.5027046e+02]
47 [-1.7089844e-03 -1.4552046e+02]
48 [-1.9531250e-03 -1.4051744e+02]
49 [ 1.6479492e-03 -1.3527052e+02]
50 [ 2.4414062e-04 -1.2941704e+02]
51 [-3.05175781e-04 -1.23411545e+02]
52 [ 3.9672852e-04 -1.1720087e+02]
53 [ 2.13623047e-03 -1.10537766e+02]
54 [ 6.1035156e-04 -1.0399693e+02]
55 [ 3.3569336e-04 -9.7202988e+01]
56 [-1.739502e-03 -9.027710e+01]
57 [-4.272461e-04 -8.335503e+01]
58 [-1.1901855e-03 -7.6376801e+01]
59 [-4.8828125e-04 -6.9428574e+01]
60 [-7.6293945e-04 -6.2581680e+01]
61 [-8.8500977e-04 -5.6045334e+01]
62 [-1.5869141e-03 -4.9427963e+01]
63 [-9.7656250e-04 -4.3270126e+01]
64 [-1.5869141e-03 -3.7204178e+01]
65 [-1.4343262e-03 -3.1393860e+01]
66 [-1.7395020e-03 -2.6090973e+01]
67 [-5.798340e-04 -2.104535e+01]
68 [-8.5449219e-04 -1.6401184e+01]
69 [-1.0070801e-03 -1.2360779e+01]
70 [ 7.9345703e-04 -8.6397400e+00]
71 [ 7.324219e-04 -5.603607e+00]
72 [-2.5939941e-03 -3.2775574e+00]
73 [ 2.4414062e-04 -1.4170227e+00]
74 [ 3.6621094e-04 -3.7210083e-01]
75 [0. 0.]
76 [0. 0.]
77 [0. 0.]
78 [0. 0.]
79 [0. 0.]
80 [0. 0.]
81 [0. 0.]
82 [0. 0.]
83 [0. 0.]
84 [0. 0.]
85 [0. 0.]
86 [0. 0.]
87 [0. 0.]
88 [0. 0.]
89 [0. 0.]
90 [0. 0.]
91 [0. 0.]
92 [0. 0.]
93 [0. 0.]
94 [0. 0.]
95 [0. 0.]
96 [0. 0.]
97 [0. 0.]
98 [0. 0.]
99 [0. 0.]
100 [0. 0.]
101 [0. 0.]
102 [0. 0.]
103 [0. 0.]
104 [0. 0.]
105 [0. 0.]
106 [0. 0.]
107 [0. 0.]
108 [0. 0.]
109 [0. 0.]
110 [0. 0.]
111 [0. 0.]
112 [0. 0.]
113 [0. 0.]
114 [0. 0.]
115 [0. 0.]
116 [0. 0.]
117 [0. 0.]
118 [0. 0.]
119 [0. 0.]
120 [0. 0.]
121 [0. 0.]
+17
View File
@@ -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