From 2baabf4ae158084abc0e01a119da2e7f4ccbf275 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lauri=20V=C3=B5sandi?= Date: Thu, 12 Sep 2024 09:53:42 +0300 Subject: [PATCH] Initial commit --- .github/workflows/docker.yml | 49 ++++++++++++++++++++++++++++++++++++ Dockerfile | 7 ++++++ 2 files changed, 56 insertions(+) create mode 100644 .github/workflows/docker.yml create mode 100644 Dockerfile diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml new file mode 100644 index 0000000..54e8667 --- /dev/null +++ b/.github/workflows/docker.yml @@ -0,0 +1,49 @@ +name: ci + +on: + push: + branches: + - main + +jobs: + docker: + strategy: + matrix: + include: + - version: v8.2.0 + model: yolov8n + - version: v8.2.0 + model: yolov8s + - version: v8.2.0 + model: yolov8m + - version: v8.2.0 + model: yolov8l + - version: v8.2.0 + model: yolov8x + + runs-on: ubuntu-latest + steps: + - + name: Checkout + uses: actions/checkout@v3 + with: + submodules: recursive + - + name: Set up Docker Buildx + uses: docker/setup-buildx-action@v2 + - + name: Login to Docker Hub + uses: docker/login-action@v2 + with: + username: ${{ github.repository_owner }} + password: ${{ secrets.DOCKERHUB_TOKEN }} + - + name: Build and push + uses: docker/build-push-action@v4 + with: + context: . + push: true + build-args: | + model=${{ matrix.model }} + version=${{ matrix.version }} + tags: ${{ github.repository }}:${{ matrix.version }}-${{ matrix.model }} diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..890bf66 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,7 @@ +ARG model=yolov8n +ARG version=v8.2.0 +FROM alpine/curl AS build +RUN curl -O https://github.com/ultralytics/assets/releases/download/${version}/${model}.pt + +FROM scratch +COPY --from=build ${model}.pt /model.pt