diff --git a/Default_1.csv b/Default_1.csv new file mode 100644 index 0000000..6c27080 --- /dev/null +++ b/Default_1.csv @@ -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.] diff --git a/Default_2.csv b/Default_2.csv new file mode 100644 index 0000000..bdda634 --- /dev/null +++ b/Default_2.csv @@ -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.] diff --git a/ocv_corner_tracker.py b/ocv_corner_tracker.py index 6c8f7f0..0f90d7a 100644 --- a/ocv_corner_tracker.py +++ b/ocv_corner_tracker.py @@ -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