Added noise removal and barycenter search

gpisanelli · Mar 13, 2020 · 37ba55e · 37ba55e
1 parent 6155ae8
commit 37ba55e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/src/main.py b/src/main.py
@@ -34,7 +34,6 @@ def precompute_box_features():
 
 def preprocess_box(b):
     pr_box = b.copy()
-    # Preprocessing (box image)
     pr_box = image_processing.convert_grayscale(pr_box)
     pr_box = image_processing.equalize_histogram(pr_box)
     if pr_box.shape[0] >= 300:

diff --git a/src/prove.py b/src/prove.py
@@ -151,7 +151,7 @@ def prove():
         cv2.circle(result, c, 20, (0,0,255), 3)
         cv2.circle(result, c, 3, (0,255,0), -1)
 
-    #visualization.display_img(image_processing.resize_img(barycenter_accumulator.astype(np.uint8), 2))
-    visualization.display_img(image_processing.resize_img(result, 2))
+    visualization.display_img(result)
 
-prove()
+
+prove()