logic will be fixed

app.py

@@ -2,7 +2,7 @@ import os
 
 import streamlit as st
 
-from utils import get_configs, get_display_names
+from utils import get_configs, get_display_names, get_path_for_viz
 
 # st.header("EVREAL  - Event-based Video Reconstruction Evaluation and Analysis Library")
 #
@@ -18,6 +18,8 @@ from utils import get_configs, get_display_names
 
 st.title("Result Analysis Tool")
 
+data_base_path = "/home/bercan/ebv/evreal_data"
+
 dataset_cfg_path = os.path.join("cfg", "dataset")
 model_cfg_path = os.path.join("cfg", "model")
 metric_cfg_path = os.path.join("cfg", "metric")
@@ -26,12 +28,12 @@ viz_cfg_path = os.path.join("cfg", "viz")
 datasets = get_configs(dataset_cfg_path)
 models = get_configs(model_cfg_path)
 metrics = get_configs(metric_cfg_path)
-vizs = get_configs(viz_cfg_path)
+visualizations = get_configs(viz_cfg_path)
 
 dataset_display_names = get_display_names(datasets)
 model_display_names = get_display_names(models)
 metric_display_names = get_display_names(metrics)
-viz_display_names = get_display_names(vizs)
+viz_display_names = get_display_names(visualizations)
 
 assert len(set(dataset_display_names)) == len(dataset_display_names), "Dataset display names are not unique"
 assert len(set(model_display_names)) == len(model_display_names), "Model display names are not unique"
@@ -57,13 +59,46 @@ selected_metric_names = st.multiselect('Select metrics to display', usable_metri
 selected_metrics = [metric for metric in usable_metrics if metric['display_name'] in selected_metric_names]
 
 selected_viz = st.multiselect('Select other visualizations to display', viz_display_names)
-selected_visualizations = [viz for viz in vizs if viz['display_name'] in selected_viz]
+selected_visualizations = [viz for viz in visualizations if viz['display_name'] in selected_viz]
 
 if not st.button('Get Results'):
     st.stop()
 
-
-# num_gt_column = 1 if len(x.intersection(y)) > 0 else 0
-#
-# num_columns = len(selected_pretrained_model_names) + 1
-
+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']
+
+recon_viz = {"name": "recon", "display_name": "Reconstruction", "viz_type": "both"}
+ground_truth = {"name": "gt", "display_name": "Ground Truth", "model_id": "groundtruth"}
+
+model_viz = [recon_viz] + both_viz + selected_metrics + model_only_viz
+
+num_columns = len(selected_models) + 1
+if len(gt_only_viz) > 0:
+    num_columns += 1
+    selected_models.append(ground_truth)
+
+num_rows = max(len(selected_metrics) + 1, len(gt_only_viz)) + 1 + len(both_viz)
+
+for row_idx in range(num_rows):
+    row_visualizations = []
+    for col_idx in range(num_columns):
+        if row_idx == 0 and col_idx == 0:
+            print("meta")
+            pass
+        elif row_idx == 0:
+            # model names
+            print(selected_models[col_idx - 1]['display_name'])
+            pass
+        elif col_idx == 0:
+            # metric names
+            print(model_viz[row_idx - 1]['display_name'])
+            pass
+        elif col_idx == num_columns - 1:
+            # gt only viz
+            print(gt_only_viz[row_idx - 1]['display_name'])
+            pass
+        else:
+            video_path = get_path_for_viz(data_base_path, selected_dataset, selected_sequence, selected_models[col_idx - 1], model_viz[row_idx - 1])
+            print(video_path)
+            pass

utils.py

@@ -29,10 +29,10 @@ def get_display_names(configs):
         display_names.append(config['display_name'])
     return display_names
 
-# def get_sequence_names(data_configs_path, dataset_name):
-#     sequences = []
-#     dataset_file_name = os.path.join(data_configs_path, dataset_name + ".json")
-#     dataset_config = read_json(dataset_file_name)
-#     for sequence in dataset_config['sequences']:
-#         sequences.append(sequence)
-#     return sequences
+
+def get_path_for_viz(base_path, dataset, sequence, model, viz):
+    dat_seq = "_".join([dataset["name"], sequence])
+    video_path = os.path.join(base_path, model['model_id'], dat_seq, "videos", viz["name"] + ".mp4")
+    return video_path
+
+