addedtitle

biomap/helper.py

@@ -63,7 +63,8 @@ def inference_on_location(model, longitude=2.98, latitude=48.81, start_date=2020
     
     images = [np.asarray(img) for img in imgs]
     labeled_imgs = [np.asarray(img) for img in labeled_imgs]
-    fig = plot_imgs_labels(dates, images, labeled_imgs, scores_details, scores)
+    title=f"TimeLapse at location {tuple(location)} between {start_date} and {end_date}"
+    fig = plot_imgs_labels(dates, images, labeled_imgs, scores_details, scores, title=title)
     # fig.save("test.png")
     return fig
 
@@ -96,7 +97,9 @@ def inference_on_location_and_month(model, longitude = 2.98, latitude = 48.81, s
     score, score_details = compute_biodiv_score(outputs[0]["linear_preds"].detach().numpy())
     logging.info(f"Calculated Biodiversity Score : {score}")
     img, label, labeled_img = transform_to_pil(outputs[0])
-    fig = plot_image([start_date], [np.asarray(img)], [np.asarray(labeled_img)], [score_details], [score])
+    
+    title=f"Prediction at location {tuple(location)} at {start_date}"
+    fig = plot_image([start_date], [np.asarray(img)], [np.asarray(labeled_img)], [score_details], [score],title=title)
     return fig
 
 

biomap/streamlit_app.py

@@ -19,7 +19,7 @@ MIN_YEAR = 2018
 MAX_YEAR = 2024
 
 FOLIUM_WIDTH = 925
-FOLIUM_HEIGHT = 600
+FOLIUM_HEIGHT = 500
 
 
 st.set_page_config(layout="wide")

biomap/utils.py

@@ -82,7 +82,7 @@ def compute_biodiv_score(class_image):
     }
     return score, score_details
 
-def plot_image(months, imgs, imgs_label, nb_values, scores):
+def plot_image(months, imgs, imgs_label, nb_values, scores, title="Single Date"):
     fig2 = px.imshow(np.array(imgs), animation_frame=0, binary_string=True)
     fig3 = px.imshow(np.array(imgs_label), animation_frame=0, binary_string=True)
     
@@ -146,9 +146,11 @@ def plot_image(months, imgs, imgs_label, nb_values, scores):
                             },)
     fig.update_xaxes(row=1, col=2, visible=False)
     fig.update_yaxes(row=1, col=2, visible=False)
+    fig.update_layout(title=title, title_x=0.5)
+
     return fig
 
-def plot_imgs_labels(months, imgs, imgs_label, nb_values, scores) :       
+def plot_imgs_labels(months, imgs, imgs_label, nb_values, scores, title="TimeLapse") :       
     fig2 = px.imshow(np.array(imgs), animation_frame=0, binary_string=True)
     fig3 = px.imshow(np.array(imgs_label), animation_frame=0, binary_string=True)
     
@@ -323,6 +325,7 @@ def plot_imgs_labels(months, imgs, imgs_label, nb_values, scores) :
 
 
     fig.update_layout(margin=dict(b=0, r=0))
+    fig.update_layout(title=title, title_x=0.5)
     return fig