Make installable via pip; fixed csv output bug.

Author: justinkay

__init__.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):

demo.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

iou_tracker/__init__.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):

iou_tracker.py iou_tracker/iou_tracker.pyiou_tracker.py renamed to iou_tracker/iou_tracker.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):

util.py iou_tracker/util.pyutil.py renamed to iou_tracker/util.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',
+)