add meta code, disabled for now

app.py

@@ -3,7 +3,7 @@ import subprocess
 
 import streamlit as st
 
-from utils import get_configs, get_display_names, get_path_for_viz, get_text_str
+from utils import get_configs, get_display_names, get_path_for_viz, get_text_str, get_meta_path
 
 # st.header("EVREAL  - Event-based Video Reconstruction Evaluation and Analysis Library")
 #
@@ -20,6 +20,12 @@ from utils import get_configs, get_display_names, get_path_for_viz, get_text_str
 
 @st.cache_data(show_spinner="Retrieving results...")
 def retrieve_results(selected_dataset, selected_sequence, selected_models, selected_metrics, selected_visualizations):
+
+    meta_enabled = False
+    meta_path = get_meta_path(base_data_dir, selected_dataset, selected_sequence)
+    if not os.path.isfile(meta_path):
+        raise ValueError("Meta file not found: {}".format(meta_path))
+
     gt_only_viz = [viz for viz in selected_visualizations if viz['viz_type'] == 'gt_only']
     model_only_viz = [viz for viz in selected_visualizations if viz['viz_type'] == 'model_only']
     both_viz = [viz for viz in selected_visualizations if viz['viz_type'] == 'both']
@@ -47,6 +53,8 @@ def retrieve_results(selected_dataset, selected_sequence, selected_models, selec
 
     padding = 2
     font_size = 20
+    meta_width = 250
+    meta_height = 70
     num_cols = len(selected_models)
     crop_str = "crop=trunc(iw/2)*2-2:trunc(ih/2)*2,"
     pad_str = "pad=ceil(iw/2)*2+{}:ceil(ih/2)*2+{}:{}:{}:white".format(padding*2, padding*2, padding, padding)
@@ -63,6 +71,15 @@ def retrieve_results(selected_dataset, selected_sequence, selected_models, selec
         left_pad = (font_size*0.7)*max([len(viz['display_name']) for viz in model_viz[1:]]) + padding*2
     else:
         left_pad = 0
+
+    if meta_enabled:  # add meta video
+        if left_pad < meta_width:
+            left_pad = meta_width
+        video_paths.append(meta_path)
+        xstack_input_parts.append("[0:v]")
+        meta_h_offset = (h - meta_height) / 2
+        layout_parts.append("0_{}".format(meta_h_offset))
+
     for row_idx in range(num_rows):
         for col_idx in range(num_cols):
             vid_idx = len(video_paths)

utils.py

@@ -37,6 +37,12 @@ def get_path_for_viz(basedir, dataset, sequence, model, viz):
     return video_path
 
 
+def get_meta_path(basedir, dataset, sequence):
+    dat_seq = "_".join([dataset["name"], sequence])
+    meta_path = os.path.join(basedir, "meta", dat_seq, "videos", "meta.mp4")
+    return meta_path
+
+
 def get_text_str(height, width, text, fontsize, fontcolor="black"):
     return "drawtext=text='{}':" \
            "fontsize={}:fontcolor={}:" \