add: docker

.gitignore

@@ -14,7 +14,6 @@ __pycache__
 Untitled*
 experiments
 third_party/REKD
-Dockerfile
 hloc/matchers/dedode.py
 gradio_cached_examples
 

Dockerfile

@@ -0,0 +1,26 @@
+# Use an official conda-based Python image as a parent image
+FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
+
+LABEL maintainer vincentqyw
+ARG PYTHON_VERSION=3.10.4
+# Set the working directory to /code
+WORKDIR /code
+# Copy the current directory contents into the container at /code
+COPY . /code
+
+RUN conda create -n imw python=${PYTHON_VERSION}
+RUN echo "source activate imw" > ~/.bashrc
+ENV PATH /opt/conda/envs/imw/bin:$PATH
+
+# Make RUN commands use the new environment
+SHELL ["conda", "run", "-n", "imw", "/bin/bash", "-c"]
+RUN pip install --upgrade pip
+
+#RUN pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
+RUN pip install torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu118
+RUN pip install -r env-docker.txt
+
+RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
+
+# Export port
+EXPOSE 7860

app.py

@@ -79,7 +79,7 @@ def ui_reset_state(*args):
 
 
 # "footer {visibility: hidden}"
-def run(config):
+def run(server_name="127.0.0.1", server_port=7860):
     """
     Runs the application.
 
@@ -329,18 +329,26 @@ def run(config):
                 ],
                 outputs=[output_wrapped, geometry_result],
             )
-
-    app.queue().launch(share=False)
+    app.queue().launch(
+        server_name=server_name, server_port=server_port, share=False
+    )
 
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument(
-        "--config_path",
+        "--server_name",
         type=str,
-        default="config.yaml",
-        help="configuration file path",
+        default="127.0.0.1",
+        help="server name",
+    )
+    parser.add_argument(
+        "--server_port",
+        type=int,
+        default=7860,
+        help="server port",
     )
+
     args = parser.parse_args()
-    config = None
-    run(config)
+    run(args.server_name, args.server_port)
+

build_docker.sh

@@ -0,0 +1,3 @@
+docker build -t image-matching-webui:latest . --no-cache
+docker tag image-matching-webui:latest vincentqin/image-matching-webui:latest
+docker push vincentqin/image-matching-webui:latest

env-docker.txt

@@ -0,0 +1,31 @@
+e2cnn==0.2.3
+einops==0.6.1
+gdown==4.7.1
+gradio==4.25.0
+gradio_client==0.15.0
+h5py==3.9.0
+imageio==2.31.1
+Jinja2==3.1.2
+kornia==0.6.12
+loguru==0.7.0
+matplotlib==3.7.1
+numpy==1.23.5
+omegaconf==2.3.0
+opencv-contrib-python==4.6.0.66
+opencv-python==4.6.0.66
+pandas==2.0.3
+plotly==5.15.0
+protobuf==4.23.2
+pycolmap==0.5.0
+pytlsd==0.0.2
+pytorch-lightning==1.4.9
+PyYAML==6.0
+scikit-image==0.21.0
+scikit-learn==1.2.2
+scipy==1.11.1
+seaborn==0.12.2
+shapely==2.0.1
+tensorboardX==2.6.1
+torchmetrics==0.6.0
+tqdm==4.65.0
+yacs==0.1.8

run_docker.sh

@@ -0,0 +1 @@
+docker run -it -p 7860:7860 vincentqin/image-matching-webui:latest python app.py --server_name "0.0.0.0" --server_port=7860