Optimization extra etsardou#2

dimitraka · Jul 20, 2020 · b0bd2d0 · b0bd2d0
1 parent 437d071
commit b0bd2d0
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/art_autonomous_exploration/src/utilities.py b/art_autonomous_exploration/src/utilities.py
@@ -74,11 +74,6 @@ def thinning(skeleton, ogml):
         for j in range(skeleton.shape[1]):
           skeleton[i][j] = yi[i][j]
 
-      # Extra Challenge #2
-      # Optimize target selection calculations
-      # itime = time.time()
-      # skeleton = np.array(y)
-
       Print.art_print("Skeletonization final copy: " + str(time.time() - itime), Print.BLUE)
       return skeleton
 
@@ -100,8 +95,8 @@ def prune(skeleton, ogml, iterations):
 
       # TODO: Must be faster!
       for i in range(skeleton.shape[0]):
-        for j in range(skeleton.shape[1]):
-          skeleton[i][j] = yi[i][j]
+         for j in range(skeleton.shape[1]):
+           skeleton[i][j] = yi[i][j]
       Print.art_print("Pruning time: " + str(time.time() - itime), Print.BLUE)
       return skeleton