Skip to content

build-agent-images

build-agent-images #9

# Copyright (C) CVAT.ai Corporation
#
# SPDX-License-Identifier: MIT
name: Build and Push Agent Images
on:
repository_dispatch:
types: [build-agent-images]
jobs:
yolo-cpu:
uses: ./.github/workflows/build-push-agent-images-reusable.yml
with:
repo_name: yolo_agent
dockerfile: detector/yolo/agent/Dockerfile
platforms: linux/amd64,linux/arm64
tag: ${{ github.event.client_payload.tag_name }}
secrets: inherit
yolo-gpu:
uses: ./.github/workflows/build-push-agent-images-reusable.yml
with:
repo_name: yolo_agent
dockerfile: detector/yolo/agent/Dockerfile
platforms: linux/amd64
build_args: USE_GPU=true
tag: ${{ github.event.client_payload.tag_name }}
tag_suffix: _GPU
secrets: inherit
sam2-cpu:
uses: ./.github/workflows/build-push-agent-images-reusable.yml
with:
repo_name: sam2_agent
dockerfile: tracker/sam2/agent/Dockerfile
platforms: linux/amd64,linux/arm64
tag: ${{ github.event.client_payload.tag_name }}
secrets: inherit
sam2-gpu:
uses: ./.github/workflows/build-push-agent-images-reusable.yml
with:
repo_name: sam2_agent
dockerfile: tracker/sam2/agent/Dockerfile
platforms: linux/amd64
build_args: USE_GPU=true
tag: ${{ github.event.client_payload.tag_name }}
tag_suffix: _GPU
secrets: inherit
transformers-cpu:
uses: ./.github/workflows/build-push-agent-images-reusable.yml
with:
repo_name: transformers_agent
dockerfile: detector/transformers/agent/Dockerfile
platforms: linux/amd64,linux/arm64
tag: ${{ github.event.client_payload.tag_name }}
secrets: inherit
transformers-gpu:
uses: ./.github/workflows/build-push-agent-images-reusable.yml
with:
repo_name: transformers_agent
dockerfile: detector/transformers/agent/Dockerfile
platforms: linux/amd64
build_args: USE_GPU=true
tag: ${{ github.event.client_payload.tag_name }}
tag_suffix: _GPU
secrets: inherit