From 712567bca7689899fda442a4b16c3ac90b8f7bc4 Mon Sep 17 00:00:00 2001
From: manurare <mreyarea@gmail.com>
Date: Tue, 30 Aug 2022 12:52:14 +0100
Subject: [PATCH] Download BoostingMonodepth weights in Dockerfile
---
Dockerfile | 8 ++++++++
code/python/src/utility/depthmap_utils.py | 1 +
2 files changed, 9 insertions(+)
diff --git a/Dockerfile b/Dockerfile
index b293993..d0441fb 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -22,6 +22,7 @@ RUN apt-get install --no-install-recommends \
libceres-dev \
python3-pybind11 \
git \
+ wget \
libboost1.71-dev -y
# 2) set up python module's build environment
@@ -33,7 +34,14 @@ RUN apt install --no-install-recommends \
# Put everything in some subfolder
WORKDIR "/monodepth"
COPY . ./
+
+# Setup BoostingMonocularDepth
RUN git submodule update --init
+RUN cd ./BoostingMonocularDepth/pix2pix/ && mkdir -p checkpoints/mergemodel
+# Midas weights
+RUN wget -P ./BoostingMonocularDepth/midas https://drive.google.com/file/d/1nqW_Hwj86kslfsXR7EnXpEWdO2csz1cC
+# Merge net weights
+RUN wget -P ./BoostingMonocularDepth/pix2pix/checkpoints/mergemodel https://sfu.ca/~yagiz/CVPR21/latest_net_G.pth
RUN pip3 install -r ./code/python/requirements.txt
diff --git a/code/python/src/utility/depthmap_utils.py b/code/python/src/utility/depthmap_utils.py
index 38a8c0f..b26f790 100644
--- a/code/python/src/utility/depthmap_utils.py
+++ b/code/python/src/utility/depthmap_utils.py
@@ -325,6 +325,7 @@ def boosting_monodepth(rgb_image_data_list):
currfile_dir = os.path.dirname(__file__)
boost_path = f"{os.path.join(currfile_dir, os.pardir, os.pardir, os.pardir, os.pardir, 'BoostingMonocularDepth')}"
sys.path.append(os.path.abspath(boost_path))
+ sys.path.append(os.path.abspath(os.path.dirname(boost_path)))
# This import fixes relative imports in subfiles within BoostingMonocularDepth project
sys.path.append(os.path.abspath(os.path.join(boost_path, "structuredrl", "models", "syncbn")))
--
GitLab