Update app.py

app.py

@@ -240,9 +240,9 @@ def pano_depth_to_world_points(depth, location):
             d_lat = lat + j/2 * np.pi / depth.shape[0] + location["pitch"] / 180 * np.pi
             
             # Convert to cartesian coordinates
-            x = radius * np.cos(d_lon) * np.sin(d_lat) + location["lng"]
+            x = radius * np.cos(d_lon) * np.sin(d_lat) + location["lat"]
             y = radius * np.cos(d_lat)
-            z = radius * np.sin(d_lon) * np.sin(d_lat) + location["lat"]
+            z = radius * np.sin(d_lon) * np.sin(d_lat) + location["lng"]
             
             pts = np.stack([x, y, z], axis=1)
             uvs = np.stack([lon, lat], axis=1)
@@ -730,11 +730,11 @@ with gr.Blocks(css=css) as demo:
                 selected = gr.Number(elem_id="fnum", value=0, minimum=0, maximum=256, interactive=False)
                 output_frame.select(fn=select_frame, inputs=[output_mask], outputs=[output_mask, selected])
                 example_coords = """[
-                  {"lat": -2.0593128411405814e-06, "lng": 7.138250595062345e-07, "heading": 152.70303, "pitch": 2.607833999999997}, 
-                  {"lat": 1.5397763348801163e-06, "lng": 1.3651150112536925e-06, "heading": 151.12973, "pitch": 2.8672300000000064}, 
-                  {"lat": -2.1552188087525792e-05, "lng": 1.5591533522041345e-05, "heading": 151.41025, "pitch": 3.4802200000000028}, 
-                  {"lat": -3.0588534016828817e-06, "lng": 3.549113042566887e-06, "heading": 151.93391, "pitch": 2.843050000000005}, 
-                  {"lat": 2.513057799546914e-05, "lng": -2.1219586638920873e-05, "heading": 152.95769, "pitch": 4.233024999999998}
+                  {"lat": 50.07379596793083, "lng": 14.437146122950555, "heading": 152.70303, "pitch": 2.607833999999997}, 
+                  {"lat": 50.073799567020004, "lng": 14.437146774240507, "heading": 151.12973, "pitch": 2.8672300000000064}, 
+                  {"lat": 50.07377647505558, "lng": 14.437161000659017, "heading": 151.41025, "pitch": 3.4802200000000028}, 
+                  {"lat": 50.07379496839027, "lng": 14.437148958238538, "heading": 151.93391, "pitch": 2.843050000000005}, 
+                  {"lat": 50.073823157821664, "lng": 14.437124189538856, "heading": 152.95769, "pitch": 4.233024999999998}
                 ]"""
                 coords = gr.Textbox(elem_id="coords", value=example_coords, label="Coordinates", interactive=False)      
                 
@@ -879,8 +879,10 @@ with gr.Blocks(css=css) as demo:
         avg[0] = avg[0] / len(locations)
         avg[1] = avg[1] / len(locations)
         for k, location in enumerate(locations):
-            locations[k]["lat"] = measure(0, 0, (location["lat"] - avg[0]), 0) / 2.5 * 100
-            locations[k]["lng"] = measure(0, 0, 0, (location["lng"] - avg[1])) / 2.5 * 100
+            lat = location["lat"] - avg[0]
+            lng = location["lng"] - avg[1]
+            locations[k]["lat"] = measure(0, 0, lat, 0) / 2.5 * 100 * np.sign(lat)
+            locations[k]["lng"] = measure(0, 0, 0, lng) / 2.5 * 100 * np.sign(lng)
         print(locations)
             
         # Process the video and get the path of the output video