You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
148 lines
6.2 KiB
148 lines
6.2 KiB
# Ultralytics YOLO 🚀, AGPL-3.0 license |
|
# Builds ultralytics/ultralytics:latest images on DockerHub https://hub.docker.com/r/ultralytics |
|
|
|
name: Publish Docker Images |
|
|
|
on: |
|
push: |
|
branches: [main] |
|
workflow_dispatch: |
|
inputs: |
|
Dockerfile: |
|
type: boolean |
|
description: Use Dockerfile |
|
default: true |
|
Dockerfile-cpu: |
|
type: boolean |
|
description: Use Dockerfile-cpu |
|
Dockerfile-arm64: |
|
type: boolean |
|
description: Use Dockerfile-arm64 |
|
Dockerfile-jetson: |
|
type: boolean |
|
description: Use Dockerfile-jetson |
|
Dockerfile-python: |
|
type: boolean |
|
description: Use Dockerfile-python |
|
Dockerfile-conda: |
|
type: boolean |
|
description: Use Dockerfile-conda |
|
push: |
|
type: boolean |
|
description: Push images to Docker Hub |
|
default: true |
|
|
|
jobs: |
|
docker: |
|
if: github.repository == 'ultralytics/ultralytics' |
|
name: Push |
|
runs-on: ubuntu-latest |
|
strategy: |
|
fail-fast: false |
|
max-parallel: 6 |
|
matrix: |
|
include: |
|
- dockerfile: "Dockerfile" |
|
tags: "latest" |
|
platforms: "linux/amd64" |
|
- dockerfile: "Dockerfile-cpu" |
|
tags: "latest-cpu" |
|
platforms: "linux/amd64" |
|
- dockerfile: "Dockerfile-arm64" |
|
tags: "latest-arm64" |
|
platforms: "linux/arm64" |
|
- dockerfile: "Dockerfile-jetson" |
|
tags: "latest-jetson" |
|
platforms: "linux/arm64" |
|
- dockerfile: "Dockerfile-python" |
|
tags: "latest-python" |
|
platforms: "linux/amd64" |
|
# - dockerfile: "Dockerfile-conda" |
|
# tags: "latest-conda" |
|
# platforms: "linux/amd64" |
|
steps: |
|
- name: Checkout repo |
|
uses: actions/checkout@v4 |
|
with: |
|
fetch-depth: 0 # copy full .git directory to access full git history in Docker images |
|
|
|
- name: Set up QEMU |
|
uses: docker/setup-qemu-action@v3 |
|
|
|
- name: Set up Docker Buildx |
|
uses: docker/setup-buildx-action@v3 |
|
|
|
- name: Login to Docker Hub |
|
uses: docker/login-action@v3 |
|
with: |
|
username: ${{ secrets.DOCKERHUB_USERNAME }} |
|
password: ${{ secrets.DOCKERHUB_TOKEN }} |
|
|
|
- name: Retrieve Ultralytics version |
|
id: get_version |
|
run: | |
|
VERSION=$(grep "^__version__ =" ultralytics/__init__.py | awk -F"'" '{print $2}') |
|
echo "Retrieved Ultralytics version: $VERSION" |
|
echo "version=$VERSION" >> $GITHUB_OUTPUT |
|
|
|
VERSION_TAG=$(echo "${{ matrix.tags }}" | sed "s/latest/${VERSION}/") |
|
echo "Intended version tag: $VERSION_TAG" |
|
echo "version_tag=$VERSION_TAG" >> $GITHUB_OUTPUT |
|
|
|
- name: Check if version tag exists on DockerHub |
|
id: check_tag |
|
run: | |
|
RESPONSE=$(curl -s https://hub.docker.com/v2/repositories/ultralytics/ultralytics/tags/$VERSION_TAG) |
|
MESSAGE=$(echo $RESPONSE | jq -r '.message') |
|
if [[ "$MESSAGE" == "null" ]]; then |
|
echo "Tag $VERSION_TAG already exists on DockerHub." |
|
echo "exists=true" >> $GITHUB_OUTPUT |
|
elif [[ "$MESSAGE" == *"404"* ]]; then |
|
echo "Tag $VERSION_TAG does not exist on DockerHub." |
|
echo "exists=false" >> $GITHUB_OUTPUT |
|
else |
|
echo "Unexpected response from DockerHub. Please check manually." |
|
echo "exists=false" >> $GITHUB_OUTPUT |
|
fi |
|
env: |
|
VERSION_TAG: ${{ steps.get_version.outputs.version_tag }} |
|
|
|
- name: Build Image |
|
if: github.event_name == 'push' || github.event.inputs[matrix.dockerfile] == 'true' |
|
run: | |
|
docker build --platform ${{ matrix.platforms }} -f docker/${{ matrix.dockerfile }} \ |
|
-t ultralytics/ultralytics:${{ matrix.tags }} \ |
|
-t ultralytics/ultralytics:${{ steps.get_version.outputs.version_tag }} . |
|
|
|
- name: Run Tests |
|
if: (github.event_name == 'push' || github.event.inputs[matrix.dockerfile] == 'true') && matrix.platforms == 'linux/amd64' && matrix.dockerfile != 'Dockerfile-conda' # arm64 images not supported on GitHub CI runners |
|
run: docker run ultralytics/ultralytics:${{ matrix.tags }} /bin/bash -c "pip install pytest && pytest tests" |
|
|
|
- name: Run Benchmarks |
|
# WARNING: Dockerfile (GPU) error on TF.js export 'module 'numpy' has no attribute 'object'. |
|
if: (github.event_name == 'push' || github.event.inputs[matrix.dockerfile] == 'true') && matrix.platforms == 'linux/amd64' && matrix.dockerfile != 'Dockerfile' && matrix.dockerfile != 'Dockerfile-conda' # arm64 images not supported on GitHub CI runners |
|
run: docker run ultralytics/ultralytics:${{ matrix.tags }} yolo benchmark model=yolov8n.pt imgsz=160 verbose=0.318 |
|
|
|
- name: Push Docker Image with Ultralytics version tag |
|
if: (github.event_name == 'push' || (github.event.inputs[matrix.dockerfile] == 'true' && github.event.inputs.push == 'true')) && steps.check_tag.outputs.exists == 'false' && matrix.dockerfile != 'Dockerfile-conda' |
|
run: | |
|
docker push ultralytics/ultralytics:${{ steps.get_version.outputs.version_tag }} |
|
|
|
- name: Push Docker Image with latest tag |
|
if: github.event_name == 'push' || (github.event.inputs[matrix.dockerfile] == 'true' && github.event.inputs.push == 'true') |
|
run: | |
|
docker push ultralytics/ultralytics:${{ matrix.tags }} |
|
if [[ "${{ matrix.tags }}" == "latest" ]]; then |
|
t=ultralytics/ultralytics:latest-runner |
|
docker build -f docker/Dockerfile-runner -t $t . |
|
docker push $t |
|
fi |
|
|
|
- name: Notify on failure |
|
if: github.event_name == 'push' && failure() # do not notify on cancelled() as cancelling is performed by hand |
|
uses: slackapi/slack-github-action@v1.24.0 |
|
with: |
|
payload: | |
|
{"text": "<!channel> GitHub Actions error for ${{ github.workflow }} ❌\n\n\n*Repository:* https://github.com/${{ github.repository }}\n*Action:* https://github.com/${{ github.repository }}/actions/runs/${{ github.run_id }}\n*Author:* ${{ github.actor }}\n*Event:* ${{ github.event_name }}\n"} |
|
env: |
|
SLACK_WEBHOOK_URL: ${{ secrets.SLACK_WEBHOOK_URL_YOLO }}
|
|
|