diff --git a/Dockerfile b/Dockerfile index b2939936309a21771c9f22b8f161bad7c64cdbd6..d0441fb13a29322d71ad56b0fd52a3661ad1a510 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 38a8c0f6e09ba8cc86513fc3e57025b10957942b..b26f7902c9ec008af224026656f7d88b564b0b4c 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")))