add: algos

Dockerfile

@@ -1,27 +1,16 @@
-# Use an official conda-based Python image as a parent image
 FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime
 LABEL maintainer vincentqyw
-ARG PYTHON_VERSION=3.10.10
 
-# Set the working directory to /code
 WORKDIR /code
 
-# Install Git and Git LFS
-RUN apt-get update && apt-get install -y git-lfs
-RUN git lfs install
+# all together
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git-lfs ffmpeg libsm6 libxext6 && \
+    git lfs install && \
+    git clone --recursive https://github.com/Vincentqyw/image-matching-webui.git . && \
+    pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir -r requirements.txt && \
+    apt-get clean && rm -rf /var/lib/apt/lists/* && \
+    pip cache purge
 
-# Clone the Git repository
-RUN git clone --recursive https://github.com/Vincentqyw/image-matching-webui.git /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 -r requirements.txt
-RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
-
-# Export port
-EXPOSE 7860
+EXPOSE 7860 8000 8001 8265

config/config.yaml

@@ -47,13 +47,13 @@ matcher_zoo:
     matcher: dad_roma
     skip_ci: true
     dense: true
-    enable: false
+    enable: true
     info:
       name: Dad(RoMa) #dispaly name
       source: "ARXIV 2025"
       github: https://github.com/example/example
       paper: https://arxiv.org/abs/2503.07347
-      display: false
+      display: true
       efficiency: low  # low, medium, high
   minima(loftr):
     matcher: minima_loftr
@@ -68,13 +68,13 @@ matcher_zoo:
     matcher: minima_roma
     skip_ci: true
     dense: true
-    enable: false
+    enable: true
     info:
       name: MINIMA(RoMa) #dispaly name
       source: "ARXIV 2024"
       paper: https://arxiv.org/abs/2412.19412
       github: https://github.com/LSXI7/MINIMA
-      display: false
+      display: true
       efficiency: low  # low, medium, high
   omniglue:
     enable: true
@@ -88,7 +88,7 @@ matcher_zoo:
       project: https://hwjiang1510.github.io/OmniGlue
       display: true
   Mast3R:
-    enable: false
+    enable: true
     matcher: mast3r
     dense: true
     info:
@@ -123,7 +123,7 @@ matcher_zoo:
       github: https://github.com/xuelunshen/gim
       paper: https://arxiv.org/abs/2402.11095
       project: https://xuelunshen.com/gim
-      display: false
+      display: true
       efficiency: low  # low, medium, high
   RoMa:
     matcher: roma
@@ -141,14 +141,14 @@ matcher_zoo:
     matcher: dkm
     skip_ci: true
     dense: true
-    enable: false
+    enable: true
     info:
       name: DKM #dispaly name
       source: "CVPR 2023"
       github: https://github.com/Parskatt/DKM
       paper: https://arxiv.org/abs/2202.00667
       project: https://parskatt.github.io/DKM
-      display: false
+      display: true
       efficiency: low  # low, medium, high
   loftr:
     matcher: loftr

docker-compose.yml

@@ -0,0 +1,41 @@
+# docker-compose.yml
+version: '3.8'
+
+services:
+  # API
+  api:
+    build: .
+    ports:
+      - "8000:8000"
+      - "8001:8001"
+      - "8265:8265"
+    command: python -m imcui.api.server --config imcui/api/config/api.yaml
+    volumes:
+      - ./models:/code/models  # Optinal: models
+    environment:
+      - PYTHONUNBUFFERED=1
+      - CUDA_VISIBLE_DEVICES=0
+
+  # Web UI
+  webui:
+    build: .
+    ports:
+      - "7860:7860"
+    command: python app.py --server_port=7860 --server_name=0.0.0.0
+    volumes:
+      - ./uploads:/code/uploads  # Optinal: some uploads
+    environment:
+      - PYTHONUNBUFFERED=1
+      - CUDA_VISIBLE_DEVICES=0
+
+  # enter debug mode
+  debug:
+    build: .
+    ports:
+      - "7860:7860"
+    command: sleep infinity
+    volumes:
+      - ./uploads:/code/uploads  # Optinal: some uploads
+    environment:
+      - PYTHONUNBUFFERED=1
+      - CUDA_VISIBLE_DEVICES=0