|
|
|
@ -6,6 +6,16 @@ name: Publish Docker Images
|
|
|
|
|
on:
|
|
|
|
|
push:
|
|
|
|
|
branches: [main]
|
|
|
|
|
workflow_dispatch:
|
|
|
|
|
inputs:
|
|
|
|
|
image:
|
|
|
|
|
type: choice
|
|
|
|
|
description: Select Docker Image
|
|
|
|
|
options:
|
|
|
|
|
- Arm64
|
|
|
|
|
- Jetson
|
|
|
|
|
- CPU
|
|
|
|
|
- GPU
|
|
|
|
|
|
|
|
|
|
jobs:
|
|
|
|
|
docker:
|
|
|
|
@ -29,6 +39,7 @@ jobs:
|
|
|
|
|
password: ${{ secrets.DOCKERHUB_TOKEN }}
|
|
|
|
|
|
|
|
|
|
- name: Build and push arm64 image
|
|
|
|
|
if: github.event_name == 'push' || github.event.inputs.image == 'Arm64'
|
|
|
|
|
uses: docker/build-push-action@v4
|
|
|
|
|
continue-on-error: true
|
|
|
|
|
with:
|
|
|
|
@ -39,6 +50,7 @@ jobs:
|
|
|
|
|
tags: ultralytics/ultralytics:latest-arm64
|
|
|
|
|
|
|
|
|
|
- name: Build and push Jetson image
|
|
|
|
|
if: github.event_name == 'push' || github.event.inputs.image == 'Jetson'
|
|
|
|
|
uses: docker/build-push-action@v4
|
|
|
|
|
continue-on-error: true
|
|
|
|
|
with:
|
|
|
|
@ -49,6 +61,7 @@ jobs:
|
|
|
|
|
tags: ultralytics/ultralytics:latest-jetson
|
|
|
|
|
|
|
|
|
|
- name: Build and push CPU image
|
|
|
|
|
if: github.event_name == 'push' || github.event.inputs.image == 'CPU'
|
|
|
|
|
uses: docker/build-push-action@v4
|
|
|
|
|
continue-on-error: true
|
|
|
|
|
with:
|
|
|
|
@ -58,6 +71,7 @@ jobs:
|
|
|
|
|
tags: ultralytics/ultralytics:latest-cpu
|
|
|
|
|
|
|
|
|
|
- name: Build and push GPU image
|
|
|
|
|
if: github.event_name == 'push' || github.event.inputs.image == 'GPU'
|
|
|
|
|
uses: docker/build-push-action@v4
|
|
|
|
|
continue-on-error: true
|
|
|
|
|
with:
|
|
|
|
|