Come on

setup.py

@@ -2,12 +2,12 @@
 
 setuptools.setup(
         name="centroid_tracker",
-        version="0.0.1",
+        version="0.0.3",
         author="Ross Leitch",
         author_email="[email protected]",
         description="A small centroid tracker library based off https://www.pyimagesearch.com/2018/07/23/simple-object-tracking-with-opencv/",
         url="https://github.com/balbatross/centroid-tracker",
-        packages=['centroid-tracker'],
+        packages=['tracker'],
         install_requires=[
             'scipy',
             'numpy'

centroid-tracker/centroidtracker.py → tracker/centroidtracker.py

@@ -2,6 +2,7 @@
 from collections import OrderedDict
 import numpy as np
 
+
 class CentroidTracker():
     def __init__(self, maxDisappeared=50):
         self.nextObjectID = 0
@@ -19,6 +20,25 @@ def deregister(self, objectID):
         del self.objects[objectID]
         del self.disappeared[objectID]
 
+    def get_id(self, rect):
+        (x, y, eX, eY) = rect
+        cX = ((x + eX) / 2.0)
+        cY = ((y + eY) / 2.0)
+
+        objectIDs = list(self.objects.keys())
+        objectCentroids = list(self.objects.values())
+
+        D = dist.cdist(np.array(objectCentroids), [(cX, cY)])
+
+        rows = D.min(axis=1).argsort()
+        cols = D.argmin(axis=1)[rows]
+        objectID = None
+
+        for (row, col) in zip(rows, cols):
+            objectID = objectIDs[row]
+            break
+        return objectID
+
     def update(self, rects):
 
         if(len(rects) == 0):