Download BoostingMonodepth weights in Dockerfile

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
 

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")))