evaluation: Move files (#3)

* evaluation: Move over files

* devcontainer: Update

.devcontainer/desktop/Dockerfile

@@ -4,16 +4,17 @@ ARG FLIRIMAGING_GID=1001
 ARG TZ=Europe/Zurich
 ARG VGLUSERS_GID=1004
 ENV DEBIAN_FRONTEND=noninteractive
-ENV LANG="en_US.UTF-8"
-ENV LANGUAGE="en_US:en"
-ENV LC_ALL="en_US.UTF-8"
 ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/opt/hpcx/ompi/lib"
 ENV PATH="${PATH}:/opt/hpcx/ompi/bin"
 
 # All apt packages
 COPY devcontainer-all-packages.sh /tmp/devcontainer-all-packages.sh
 RUN /tmp/devcontainer-all-packages.sh && rm -f /tmp/devcontainer-all-packages.sh
 
+ENV LANG="en_US.UTF-8" \
+    LANGUAGE="en_US:en" \
+    LC_ALL="en_US.UTF-8"
+
 # Python packages
 COPY requirements.txt /tmp/requirements.txt
 RUN python3 -m pip install --upgrade pip && \

.devcontainer/desktop/devcontainer.json

@@ -3,8 +3,7 @@
     "customizations": {
         "vscode": {
             "extensions": [
-                "ms-python.python",
-                "github.vscode-github-actions"
+                "ms-python.python"
             ],
             "settings": {
                 "files.hotExit": "off",
@@ -18,7 +17,7 @@
     },
     "remoteUser": "asl",
     "initializeCommand": ".devcontainer/devcontainer-optional-mounts.sh",
-    "postStartCommand": "/bin/bash",
+    "postStartCommand": "pip install -e .",
     "mounts": [
         {
             "source": "${localEnv:HOME}/.bash-git-prompt",

.devcontainer/devcontainer-all-packages.sh

@@ -10,6 +10,7 @@ main() {
         build-essential
         ca-certificates
         ccache
+        cm-super
         curl
         gawk
         gnupg

.github/workflows/ci.yml

@@ -12,7 +12,7 @@ jobs:
 
     strategy:
       matrix:
-        ci_script: [pr_unittest]
+        ci_script: [pr_unittest, pr_evaluate_networks, pr_train_networks]
 
     steps:
       - name: Checkout

.gitignore

@@ -160,3 +160,7 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
+
+tests/resources/results
+tests/resources/*.png
+tests/resources/*.pkl

ci/pr_evaluate_networks.bash

@@ -0,0 +1,21 @@
+#!/bin/bash
+
+# Install radarmeetsvision
+pip install -e .
+
+RESULTS_DIR=tests/resources/results
+if [ -d $RESULTS_DIR ]; then
+    rm -r $RESULTS_DIR
+fi
+rm -rf tests/resources/*.pkl
+
+# Run the evaluation script
+python3 scripts/evaluation/evaluate_networks.py --dataset tests/resources --config tests/resources/test_evaluation.json --output tests/resources --network tests/resources
+
+TEX_FILE="tests/resources/results/results_table0.tex"
+if [[ -f "$TEX_FILE" && -s "$TEX_FILE" ]]; then
+    echo "Evaluation script successful, .tex table exists and is not empty."
+else
+    echo "Evaluation script failed, .tex table does not exist or is empty."
+    exit 1
+fi

ci/pr_train_networks.bash

@@ -0,0 +1,55 @@
+#!/bin/bash
+
+checkpoints=tests/resources
+save_path=tests/resources
+datasets=tests/resources
+
+config_path=tests/resources
+config_radar=$config_path/test_train_radar.json
+config_metric=$config_path/test_train_metric.json
+config_relative=$config_path/test_train_relative.json
+
+# Install radarmeetsvision
+pip install -e .
+
+# RADAR TRAINING (depth prior + 2 output channels)
+python3 scripts/train.py \
+--checkpoints $checkpoints \
+--config $config_radar \
+--datasets $datasets \
+--results ""
+
+if [ $? -eq 0 ]; then
+    echo "Radar training script successful"
+else
+    echo "Training script failed"
+    exit 1
+fi
+
+# RGB TRAINING (no depth prior + 1 output channel)
+python3 scripts/train.py \
+--checkpoints $checkpoints \
+--config $config_metric \
+--datasets $datasets \
+--results ""
+
+if [ $? -eq 0 ]; then
+    echo "RGB training script successful"
+else
+    echo "Training script failed"
+    exit 1
+fi
+
+# Relative RGB TRAINING (no depth prior + 1 output channel)
+python3 scripts/train.py \
+--checkpoints $checkpoints \
+--config $config_relative \
+--datasets $datasets \
+--results ""
+
+if [ $? -eq 0 ]; then
+    echo "Relative RGB training script successful"
+else
+    echo "Training script failed"
+    exit 1
+fi

ci/pr_unittest.bash

@@ -1,6 +1,6 @@
 #!/bin/bash
 
-if pip install .
+if pip install -e .
 then
     echo "Installation successful"
 else

scripts/evaluation/config.json

@@ -0,0 +1,77 @@
+{
+    "scenarios": {
+        "Industrial Hall": "maschinenhalle0",
+        "Agricultural Field": "outdoor0",
+        "Rhône Glacier": "rhone_flight"
+    },
+    "networks": {
+        "Metric Depth \\cite{depthanythingv2}-S": "rgb_s_bs8_e9.pth",
+        "Metric Depth \\cite{depthanythingv2}-B": "rgb_b_bs4_e8.pth",
+        "Scaled Relative Depth \\cite{depthanythingv2}-S": "relrgb_s_bs8_e9.pth",
+        "Scaled Relative Depth \\cite{depthanythingv2}-B": "relrgb_b_bs4_e9.pth",
+        "Ours-S": "radar_s_bs8_e19.pth",
+        "Ours-B": "radar_b_bs4_e21.pth"
+    },
+    "Metric Depth \\cite{depthanythingv2}-S": {
+        "use_depth_prior": false,
+        "output_channels": 1,
+        "relative_depth": 0,
+        "depth_min": 0.19983673095703125,
+        "depth_max": 120.49285888671875,
+        "encoder": "vits",
+        "marker": "X",
+        "plot_prediction": 0
+    },
+    "Metric Depth \\cite{depthanythingv2}-B": {
+        "use_depth_prior": false,
+        "output_channels": 1,
+        "relative_depth": 0,
+        "depth_min": 0.19983673095703125,
+        "depth_max": 120.49285888671875,
+        "encoder": "vitb",
+        "marker": "X",
+        "plot_prediction": 0
+    },
+    "Scaled Relative Depth \\cite{depthanythingv2}-S": {
+        "use_depth_prior": false,
+        "output_channels": 1,
+        "relative_depth": 1,
+        "depth_min": 0.0,
+        "depth_max": 1.0,
+        "encoder": "vits",
+        "marker": "D",
+        "plot_prediction": 0
+    },
+    "Scaled Relative Depth \\cite{depthanythingv2}-B": {
+        "use_depth_prior": false,
+        "output_channels": 1,
+        "relative_depth": 1,
+        "depth_min": 0.0,
+        "depth_max": 1.0,
+        "encoder": "vitb",
+        "marker": "D",
+        "plot_prediction": 0
+    },
+    "Ours-S": {
+        "use_depth_prior": true,
+        "output_channels": 2,
+        "relative_depth": 0,
+        "depth_min": 0.19983673095703125,
+        "depth_max": 120.49285888671875,
+        "encoder": "vits",
+        "marker": "o",
+        "plot_prediction": 0
+    },
+    "Ours-B": {
+        "use_depth_prior": true,
+        "output_channels": 2,
+        "relative_depth": 0,
+        "depth_min": 0.19983673095703125,
+        "depth_max": 120.49285888671875,
+        "encoder": "vitb",
+        "marker": "o",
+        "plot_prediction": 1
+    },
+    "height": 480,
+    "width": 640
+}

scripts/evaluation/create_scatter_plot.py

@@ -0,0 +1,98 @@
+######################################################################
+#
+# Copyright (c) 2024 ETHZ Autonomous Systems Lab. All rights reserved.
+#
+######################################################################
+
+import matplotlib as mpl
+import matplotlib.pyplot as plt
+import numpy as np
+
+from matplotlib.gridspec import GridSpec
+from pathlib import Path
+from PIL import Image
+
+mpl.rcParams['font.size'] = 8
+mpl.rcParams['figure.figsize'] = [9, 2.7]
+mpl.rcParams['lines.linewidth'] = 0.6
+mpl.rcParams['grid.linewidth'] = 0.5
+mpl.rcParams['axes.linewidth'] = 0.6
+mpl.rcParams['axes.labelsize'] = 8
+mpl.rcParams['axes.titlesize'] = 8
+mpl.rcParams['legend.fontsize'] = 8
+mpl.rcParams['xtick.labelsize'] = 8
+mpl.rcParams['ytick.labelsize'] = 8
+mpl.rcParams['text.usetex'] = True
+mpl.rcParams['font.family'] = 'serif'
+
+def create_scatter_plot(results_per_scenario_sample, config, output_dir, subsample=10):
+    scenario_color = {
+        'Industrial Hall': 'k',
+        'Agricultural Field': 'green',
+        'Rhône Glacier': 'navy'
+    }
+    label_dict = {
+        "Metric Depth \\cite{depthanythingv2}-B": "Metric Depth-B",
+        "Ours-B": "Ours-B"
+    }
+    samples = [['00000_rgb.jpg', '00000_dp.jpg'],
+               ['00050_rgb.jpg', '00050_dp.jpg'],
+               ['00250_rgb.jpg', '00250_dp.jpg']]
+
+    sample_dir = Path('scripts/evaluation/samples')
+    img_out = None
+    for i, sample in enumerate(samples):
+        rgb_file = sample_dir / sample[0]
+        dp_file = sample_dir / sample[1]
+
+        rgb = np.array(Image.open(rgb_file))
+        dp = np.array(Image.open(dp_file))
+        img = np.concatenate((rgb, dp), axis=1)
+        if img_out is None:
+            img_out = img
+        else:
+            img_out = np.concatenate((img_out, img), axis=0)
+    border=5
+    img_out[:border, :] = (0.0, 0.0, 0.0)
+    img_out[-border:, :] = (0.0, 0.0, 0.0)
+    img_out[:, :border] = (0.0, 0.0, 0.0)
+    img_out[:, -border:] = (0.0, 0.0, 0.0)
+
+
+    fig = plt.figure(figsize=(9, 2.7))  # Adjust the figure size if needed
+    gs = GridSpec(1, 2, width_ratios=[1, 2])  # Set the width ratios to 1:2
+
+    ax0 = fig.add_subplot(gs[0])
+    ax0.imshow(img_out)
+    ax0.axis('off')
+
+    ax1 = fig.add_subplot(gs[1])
+    for scenario_key in config['scenarios'].keys():
+        for i, network_key in enumerate(config['networks'].keys()):
+            if '-B' in network_key and not 'Scaled' in network_key:
+                average_depths = results_per_scenario_sample[scenario_key][network_key]['average_depth']
+                abs_rel_values = results_per_scenario_sample[scenario_key][network_key]['abs_rel']
+                average_depths_subsampled = average_depths[::subsample]
+                abs_rel_values_subsampled = abs_rel_values[::subsample]
+                label = label_dict[network_key] + ' ' + (scenario_key if 'Metric' in network_key else '')
+                ax1.scatter(average_depths_subsampled, abs_rel_values_subsampled, label=label, marker=config[network_key]['marker'], c=scenario_color[scenario_key], s=25, alpha=0.5)
+
+
+    # Set axis labels, title, and legend
+    plt.xlabel('Average Scene Depth [m]')
+    plt.ylabel('Absolute Relative Error [ ]')
+    plt.legend(loc='upper right')
+    ax1.grid()
+    plt.tight_layout()
+
+    # Save the plot
+    output_file = Path(output_dir) / f'results_overview.png'
+    plt.savefig(str(output_file), transparent=True, bbox_inches='tight', dpi=400)
+    plt.close()
+
+    # Post-process the saved image to crop any unnecessary white space
+    img = Image.open(str(output_file))
+    img = img.convert("RGBA")
+    bbox = img.getbbox()
+    cropped_img = img.crop(bbox)
+    cropped_img.save(str(output_file))