From 4135fc75f9ff499a5e3b7a35d699e84bdbbea5e3 Mon Sep 17 00:00:00 2001
From: Lars Nieradzik <l.nieradzik@gmail.com>
Date: Mon, 2 Apr 2018 21:18:20 +0200
Subject: [PATCH] small bug fix

---
 kmeans.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/kmeans.py b/kmeans.py
index 7cb17e5..9d823bc 100644
--- a/kmeans.py
+++ b/kmeans.py
@@ -42,7 +42,7 @@ def translate_boxes(boxes):
     for row in range(new_boxes.shape[0]):
         new_boxes[row][2] = np.abs(new_boxes[row][2] - new_boxes[row][0])
         new_boxes[row][3] = np.abs(new_boxes[row][3] - new_boxes[row][1])
-    return np.delete(boxes, [0, 1], axis=1)
+    return np.delete(new_boxes, [0, 1], axis=1)
 
 
 def kmeans(boxes, k, dist=np.median):
-- 
GitLab