correction, sdk version, plot size

README.md

@@ -4,6 +4,7 @@ emoji: 🚙
 colorFrom: red
 colorTo: gray
 sdk: gradio
+sdk_version: 3.8
 app_file: app.py
 pinned: false
 language: 

scripts/scripts_utils/plotly_interface.py

@@ -107,6 +107,8 @@ def build_data(
     mask_pred = numpy_data["mask_pred"][0]
     map_data = numpy_data["map_data"][0]
     mask_map = numpy_data["mask_map"][0]
+    
+    marker_size = 12
 
     data_x = get_scatter_data(
         x,
@@ -119,14 +121,14 @@ def build_data(
         x=x[0:1, -1:],
         mask_x=mask_x[0:1, -1:],
         mode="markers",
-        marker=dict(color="blue", size=20, opacity=0.5),
+        marker=dict(color="blue", size=marker_size, opacity=0.5),
         name="Ego",
     )
     agent_present = get_scatter_data(
         x=x[1:2, -1:],
         mask_x=mask_x[1:2, -1:],
         mode="markers",
-        marker=dict(color="green", size=20, opacity=0.5),
+        marker=dict(color="green", size=marker_size, opacity=0.5),
         name="Agent",
     )
     
@@ -161,7 +163,7 @@ def build_data(
             pred[:, i, -1:],
             mask_pred[:, -1:],
             mode="markers",
-            marker=dict(color="red", size=10, opacity=0.5, symbol="x"),
+            marker=dict(color="red", size=marker_size/2, opacity=0.5, symbol="x"),
             name="Forecast end",
         )
         forecasts_end += forecast_end
@@ -177,7 +179,7 @@ def build_data(
                     y=x[mask_x[:, k], k, 1],
                     mode="markers",
                     opacity=animation_opacity,
-                    marker=dict(color="black", size=15),
+                    marker=dict(color="black", size=marker_size),
                     showlegend=False,
                 ),
                 go.Scatter(
@@ -185,7 +187,7 @@ def build_data(
                     y=x[0:1, k, 1],
                     mode="markers",
                     opacity=animation_opacity,
-                    marker=dict(color="blue", size=15),
+                    marker=dict(color="blue", size=marker_size),
                     showlegend=False,
                 ),
             ]
@@ -200,14 +202,14 @@ def build_data(
             y=y[1:2][mask_y[1:2, k], k, 1],
             mode="markers",
             opacity=animation_opacity,
-            marker=dict(color="green", size=15),
+            marker=dict(color="green", size=marker_size),
         )
         cur_gt_future_data = go.Scatter(
             x=y[2:][mask_y[2:, k], k, 0],
             y=y[2:][mask_y[2:, k], k, 1],
             mode="markers",
             opacity=animation_opacity,
-            marker=dict(color="black", size=15),
+            marker=dict(color="black", size=marker_size),
         )
         cur_pred_data = []
         for i in range(n_samples):
@@ -217,7 +219,7 @@ def build_data(
                     y=pred[mask_pred[:, k], i, k, 1],
                     mode="markers",
                     opacity=animation_opacity,
-                    marker=dict(color="red", size=15),
+                    marker=dict(color="red", size=marker_size),
                     showlegend=False,
                 )
             )
@@ -226,7 +228,7 @@ def build_data(
             y=y[0:1, k, 1],
             mode="markers",
             opacity=animation_opacity,
-            marker=dict(color="blue", size=15),
+            marker=dict(color="blue", size=marker_size),
         )
         cur_data = [cur_gt_agent_data, cur_gt_future_data, *cur_pred_data, cur_ego_data]
         frame = go.Frame(data=cur_data)
@@ -262,8 +264,8 @@ def prediction_plot(
         ),
         title_text="Road Scene",
         hovermode="closest",
-        width=600,
-        height=300,
+        width=800,
+        height=400,
         updatemenus=[
             dict(
                 type="buttons",
@@ -274,7 +276,6 @@ def prediction_plot(
                         args=[
                             None,
                             dict(
-                                transition=dict(duration=100),
                                 frame=dict(duration=100, redraw=False),
                                 mode="immediate",
                                 fromcurrent=True,