Update app.py

app.py

@@ -222,6 +222,15 @@ def pano_depth_to_world_points(depth):
     latitude : -pi/2 to pi/2
     """
 
+    i = depth.shape[0]-5
+    kernel = np.ones((depth.shape[1],1),np.float32)/depth.shape[1]
+    avgs = cv2.filter2D(depth,-1,kernel)
+    while i>=2:
+        if avgs[i-1][0] < avgs[i][0] and avgs[i-1][0] < avgs[i-2][0]:
+            print(i-1)
+            break
+        i=i-1
+
     # Convert depth to radius
     radius = (255 - depth.flatten())
 
@@ -253,15 +262,6 @@ def pano_depth_to_world_points(depth):
             #print(f'i: {i}, j: {j}')
             j = j+1
         i = i+1
-
-    i = depth.shape[0]-1
-    kernel = np.ones((depth.shape[1],1),np.float32)/depth.shape[1]
-    avgs = cv2.filter2D(depth,-1,kernel)
-    while i>=2:
-        if avgs[i-1][0] < avgs[i][0] and avgs[i-1][0] < avgs[i-2][0]:
-            print(i-1)
-            break
-        i=i-1
         
     return [pts3d, uv]