Add Dockerfile

.dockerignore

@@ -0,0 +1,13 @@
+venv/
+outputs/
+modules/__pycache__/
+models/
+temp/
+
+*.mp4
+*.jpg
+*.png
+
+.idea
+.git
+.github

Dockerfile

@@ -0,0 +1,32 @@
+# Build Stage
+FROM debian:bookworm-slim AS builder
+
+RUN apt-get update && \
+    apt-get install -y curl git python3 python3-pip python3-venv && \
+    rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/* && \
+    mkdir -p /sam2-playground
+
+WORKDIR /sam2-playground
+
+COPY requirements.txt .
+
+RUN python3 -m venv venv && \
+    . venv/bin/activate && \
+    pip install --no-cache-dir -r requirements.txt
+
+# Runtime Stage
+FROM debian:bookworm-slim AS runtime
+
+RUN apt-get update && \
+    apt-get install -y curl ffmpeg python3 && \
+    rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/*
+
+WORKDIR /sam2-playground
+
+COPY . .
+COPY --from=builder /sam2-playground/venv /sam2-playground/venv
+
+ENV PATH="/sam2-playground/venv/bin:$PATH"
+ENV LD_LIBRARY_PATH="/sam2-playground/venv/lib64/python3.11/site-packages/nvidia/cublas/lib:/sam2-playground/venv/lib64/python3.11/site-packages/nvidia/cudnn/lib"
+
+ENTRYPOINT ["python", "app.py"]

docker-compose.yaml

@@ -0,0 +1,18 @@
+version: '3.8'
+
+services:
+  app:
+    build: .
+    image: jhj0517/sam2-playground:latest
+
+    volumes:
+      # Mount models and output paths to your custom paths like this, e.g:
+      # - C:/sam2-models/custom/path:/sam2-playground/models
+      # - C:/sam2-playground-outputs/custom/path:/sam2-playground/outputs
+      - /sam2-playground/models
+      - /sam2-playground/outputs
+
+    ports:
+      - "8080:8080"
+
+    entrypoint: ["python", "app.py", "--server_port", "8080", "--server_name", "0.0.0.0",]