Merge pull request #7 from moriyalab/support_new_docker_system

Support New Docker System

.github/workflows/docker.yml

@@ -0,0 +1,40 @@
+name: Docker Build and Push to GHCR
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+    branches:
+      - main
+
+jobs:
+  build-and-push:
+    runs-on: ubuntu-22.04
+
+    steps:
+      - name: Free Disk Space (Ubuntu)
+        uses: jlumbroso/free-disk-space@main
+        with:
+          tool-cache: false
+
+      - name: Checkout repository
+        uses: actions/checkout@v3
+
+      - name: Log in to GitHub Container Registry
+        if: github.event_name == 'push'
+        uses: docker/login-action@v2
+        with:
+          registry: ghcr.io
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Build Docker image
+        run: |
+          docker build -t ghcr.io/${{ github.repository }}_dev:latest --target builder_image .
+          docker build -t ghcr.io/${{ github.repository }}:latest --target release_image .
+
+      - name: Push Docker image
+        if: github.event_name == 'push'
+        run: |
+          docker push ghcr.io/${{ github.repository }}_dev:latest
+          docker push ghcr.io/${{ github.repository }}:latest

.gitignore

@@ -9,5 +9,6 @@
 *.png
 *.onnx
 *.engine
+*.mcap
 
 flagged*/

Dockerfile

@@ -0,0 +1,19 @@
+FROM ultralytics/ultralytics:8.2.63 AS builder_image
+
+RUN apt-get update && apt-get install -y git curl ffmpeg
+RUN git config --global --add safe.directory /app
+RUN python3 -m pip install --upgrade pip
+RUN python3 -m pip install poetry \
+  && poetry config virtualenvs.create false
+
+WORKDIR /app
+
+COPY pyproject.toml poetry.lock ./
+RUN poetry install --no-root
+
+FROM builder_image AS release_image
+
+COPY --from=builder_image /app /app
+COPY . /app/
+
+CMD [ "python3", "app.py" ]

poetylock.sh

@@ -1,7 +1,4 @@
 #!/bin/bash
 
-cp /home/root/docker/pyproject.toml /app/pyproject.toml
 cd /app
 poetry lock --no-update
-rm -rf /home/root/docker/poetry.lock
-cp /app/poetry.lock /home/root/docker/poetry.lock

run_dev_env.sh

@@ -32,7 +32,7 @@ PLATFORM="$(uname -m)"
 if [ $PLATFORM = "x86_64" ]; then
     echo "x86"
     docker pull ghcr.io/moriyalab/horus_inference_server:latest
-    docker run -it --rm --gpus all --runtime nvidia --shm-size=32G -v $ROOT/datasets:/usr/src/datasets -v $ROOT:/home/root -w /home/root --network host ghcr.io/moriyalab/horus_inference_server:latest
+    docker run -it --rm --gpus all --runtime nvidia --shm-size=32G -v $ROOT/datasets:/usr/src/datasets -v $ROOT:/home/root -w /home/root --network host ghcr.io/moriyalab/horus_inference_server_dev:latest
 else
     echo "Not Support Platform. Only support x86."
 fi