From e00c16ddb81aed74e4d7c78acaf89bd282187c99 Mon Sep 17 00:00:00 2001 From: justinkay Date: Sat, 28 Mar 2020 14:29:39 -0500 Subject: [PATCH 1/3] Make installable via pip; fixed csv output bug. --- __init__.py | 0 demo.py | 6 +++--- iou_tracker/__init__.py | 0 iou_tracker.py => iou_tracker/iou_tracker.py | 0 util.py => iou_tracker/util.py | 2 +- viou_tracker.py => iou_tracker/viou_tracker.py | 0 vis_tracker.py => iou_tracker/vis_tracker.py | 0 mot16.py | 4 ++-- mot17.py | 4 ++-- setup.py | 10 ++++++++++ 10 files changed, 18 insertions(+), 8 deletions(-) create mode 100644 __init__.py create mode 100644 iou_tracker/__init__.py rename iou_tracker.py => iou_tracker/iou_tracker.py (100%) rename util.py => iou_tracker/util.py (99%) rename viou_tracker.py => iou_tracker/viou_tracker.py (100%) rename vis_tracker.py => iou_tracker/vis_tracker.py (100%) create mode 100644 setup.py diff --git a/__init__.py b/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/demo.py b/demo.py index 3041970..0dabab0 100755 --- a/demo.py +++ b/demo.py @@ -9,9 +9,9 @@ import argparse -from iou_tracker import track_iou -from viou_tracker import track_viou -from util import load_mot, save_to_csv +from iou_tracker.iou_tracker import track_iou +from iou_tracker.viou_tracker import track_viou +from iou_tracker.util import load_mot, save_to_csv def main(args): diff --git a/iou_tracker/__init__.py b/iou_tracker/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/iou_tracker.py b/iou_tracker/iou_tracker.py similarity index 100% rename from iou_tracker.py rename to iou_tracker/iou_tracker.py diff --git a/util.py b/iou_tracker/util.py similarity index 99% rename from util.py rename to iou_tracker/util.py index c6717b4..d6282d7 100644 --- a/util.py +++ b/iou_tracker/util.py @@ -194,7 +194,7 @@ def save_to_csv(out_path, tracks, fmt='motchallenge'): 'y': bbox[1]+1, 'w': bbox[2] - bbox[0], 'h': bbox[3] - bbox[1], - 'score': track['max_score']} + 'score': track['max_score'].item()} if fmt == 'motchallenge': row['wx'] = -1 row['wy'] = -1 diff --git a/viou_tracker.py b/iou_tracker/viou_tracker.py similarity index 100% rename from viou_tracker.py rename to iou_tracker/viou_tracker.py diff --git a/vis_tracker.py b/iou_tracker/vis_tracker.py similarity index 100% rename from vis_tracker.py rename to iou_tracker/vis_tracker.py diff --git a/mot16.py b/mot16.py index b7bf173..25c02a2 100755 --- a/mot16.py +++ b/mot16.py @@ -10,8 +10,8 @@ from time import time import argparse -from iou_tracker import track_iou -from util import load_mot, save_to_csv +from iou_tracker.iou_tracker import track_iou +from iou_tracker.util import load_mot, save_to_csv def main(args): diff --git a/mot17.py b/mot17.py index 1da6b8d..f018e28 100755 --- a/mot17.py +++ b/mot17.py @@ -10,8 +10,8 @@ from time import time import argparse -from iou_tracker import track_iou -from util import load_mot, save_to_csv +from iou_tracker.iou_tracker import track_iou +from iou_tracker.util import load_mot, save_to_csv def main(args): diff --git a/setup.py b/setup.py new file mode 100644 index 0000000..576328d --- /dev/null +++ b/setup.py @@ -0,0 +1,10 @@ +from setuptools import find_packages, setup + +setup( + name='iou_tracker', + packages=find_packages(), + version='0.1', + description='IOU Tracker', + author='Erik Bochinski and Volker Eiselein and Thomas Sikora', + license='MIT License', +) From 6d92f407d274bb1468ab9ae07ac8c02e77889b30 Mon Sep 17 00:00:00 2001 From: justinkay Date: Wed, 1 Apr 2020 13:37:43 -0500 Subject: [PATCH 2/3] Fixed imports and made tracker 0-indexed. --- iou_tracker/viou_tracker.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/iou_tracker/viou_tracker.py b/iou_tracker/viou_tracker.py index 020a30e..81efad4 100644 --- a/iou_tracker/viou_tracker.py +++ b/iou_tracker/viou_tracker.py @@ -11,8 +11,8 @@ from tqdm import tqdm from time import time -from util import iou, load_mot -from vis_tracker import VisTracker +from .util import iou, load_mot +from .vis_tracker import VisTracker def track_viou(frames_path, detections, sigma_l, sigma_h, sigma_iou, t_min, ttl, tracker_type, keep_upper_height_ratio): @@ -45,7 +45,7 @@ def track_viou(frames_path, detections, sigma_l, sigma_h, sigma_iou, t_min, ttl, tracks_finished = [] frame_buffer = [] - for frame_num, detections_frame in enumerate(tqdm(detections), start=1): + for frame_num, detections_frame in enumerate(tqdm(detections), start=0): # load frame and put into buffer frame_path = frames_path.format(frame_num) frame = cv2.imread(frame_path) From 957697afdb94867ec29b70448fe9ca62ac127ec5 Mon Sep 17 00:00:00 2001 From: justinkay Date: Sun, 30 May 2021 21:09:36 +0000 Subject: [PATCH 3/3] Fix import and frame indexing for IOU tracker as well. --- iou_tracker/iou_tracker.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/iou_tracker/iou_tracker.py b/iou_tracker/iou_tracker.py index 12cd6bb..dcb1cea 100644 --- a/iou_tracker/iou_tracker.py +++ b/iou_tracker/iou_tracker.py @@ -7,7 +7,7 @@ from time import time -from util import load_mot, iou +from .util import load_mot, iou def track_iou(detections, sigma_l, sigma_h, sigma_iou, t_min): @@ -30,7 +30,7 @@ def track_iou(detections, sigma_l, sigma_h, sigma_iou, t_min): tracks_active = [] tracks_finished = [] - for frame_num, detections_frame in enumerate(detections, start=1): + for frame_num, detections_frame in enumerate(detections, start=0): # apply low threshold to detections dets = [det for det in detections_frame if det['score'] >= sigma_l]