commit wip

app.py

@@ -58,7 +58,13 @@ usable_metric_display_names = get_display_names(usable_metrics)
 selected_metric_names = st.multiselect('Select metrics to display', usable_metric_display_names)
 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)
+if not selected_dataset['has_frames']:
+    usable_viz = [viz for viz in visualizations if viz['gt_type'] != 'frame']
+else:
+    usable_viz = visualizations
+usable_viz_display_names = get_display_names(usable_viz)
+
+selected_viz = st.multiselect('Select other visualizations to display', usable_viz_display_names)
 selected_visualizations = [viz for viz in visualizations if viz['display_name'] in selected_viz]
 
 if not st.button('Get Results'):
@@ -68,21 +74,29 @@ gt_only_viz = [viz for viz in selected_visualizations if viz['viz_type'] == 'gt_
 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"}
+recon_viz = {"name": "recon", "display_name": "Reconstruction", "viz_type": "both", "gt_type": "frame"}
 ground_truth = {"name": "gt", "display_name": "Ground Truth", "model_id": "groundtruth"}
 
 model_viz = [recon_viz] + both_viz + selected_metrics + model_only_viz
+num_model_rows = len(model_viz) + 1
+
+gt_viz = []
+if selected_dataset['has_frames']:
+    gt_viz.append(recon_viz)
+    gt_viz.extend([viz for viz in both_viz if viz['gt_type'] == 'frame'])
+    gt_viz.extend([viz for viz in gt_only_viz if viz['gt_type'] == 'frame'])
 
-num_columns = len(selected_models) + 1
-if len(gt_only_viz) > 0:
-    num_columns += 1
-    selected_models.append(ground_truth)
+gt_viz.extend([viz for viz in both_viz if viz['gt_type'] == 'event'])
+gt_viz.extend([viz for viz in gt_viz if viz['gt_type'] == 'event'])
 
-num_rows = max(len(selected_metrics) + 1, len(gt_only_viz)) + 1 + len(both_viz)
+num_gt_rows = len(gt_viz) + 1
+num_rows = max(num_model_rows, num_gt_rows)
+
+num_model_columns = len(selected_models) + 1
 
 for row_idx in range(num_rows):
     row_visualizations = []
-    for col_idx in range(num_columns):
+    for col_idx in range(num_model_columns):
         if row_idx == 0 and col_idx == 0:
             print("meta")
             pass
@@ -94,10 +108,6 @@ for row_idx in range(num_rows):
             # 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)

cfg/dataset/BS_ERGB.json

@@ -3,7 +3,7 @@
     "width": 970,
     "height": 625,
     "no_ref": false,
-     "has_gt": true,
+    "has_frames": true,
     "sequences": 
     {
         "may29_handheld_01": {},

cfg/dataset/ECD.json

@@ -3,7 +3,7 @@
     "width": 240,
     "height": 180,
     "no_ref": false,
-    "has_gt": true,
+    "has_frames": true,
     "sequences": 
     {
         "calibration":

cfg/dataset/ECD_FAST.json

@@ -3,7 +3,7 @@
     "width": 240,
     "height": 180,
     "no_ref": true,
-    "has_gt": true,
+    "has_frames": true,
     "sequences": 
     {
         "boxes_6dof":

cfg/dataset/HDR.json

@@ -3,7 +3,7 @@
     "width": 640,
     "height": 480,
     "no_ref": true,
-    "has_gt": false,
+    "has_frames": false,
     "sequences": 
     {
         "hdr_selfie": {},

cfg/dataset/HQF.json

@@ -3,7 +3,7 @@
     "width": 240,
     "height": 180,
     "no_ref": false,
-    "has_gt": true,
+    "has_frames": true,
     "sequences":
     {
         "bike_bay_hdr": {},

cfg/dataset/MVSEC.json

@@ -3,7 +3,7 @@
     "width": 346,
     "height": 260,
     "no_ref": false,
-    "has_gt": true,
+    "has_frames": true,
     "sequences": 
     {
         "indoor_flying1_data":

cfg/dataset/MVSEC_NIGHT.json

@@ -3,7 +3,7 @@
     "width": 346,
     "height": 260,
     "no_ref": true,
-    "has_gt": true,
+    "has_frames": false,
     "sequences": 
     {
         "outdoor_night1_data": {},

cfg/viz/event_images.json

@@ -1,4 +1,5 @@
 {
     "display_name": "Event Images",
-    "viz_type": "gt_only"
+    "viz_type": "gt_only",
+    "gt_type": "event"
 }

cfg/viz/event_rate.json

@@ -1,4 +1,5 @@
 {
     "display_name": "Event Rate",
-    "viz_type": "gt_only"
+    "viz_type": "gt_only",
+    "gt_type": "event"
 }

cfg/viz/histogram.json

@@ -1,4 +1,5 @@
 {
     "display_name": "Histogram",
-    "viz_type": "both"
+    "viz_type": "both",
+    "gt_type": "frame"
 }