Skip to content

Commit

Permalink
adding a object_detection_client workflow file
Browse files Browse the repository at this point in the history
Signed-off-by: greg pereira <[email protected]>
  • Loading branch information
Gregory-Pereira committed May 2, 2024
1 parent 958fdfa commit 9397da0
Show file tree
Hide file tree
Showing 4 changed files with 100 additions and 6 deletions.
89 changes: 89 additions & 0 deletions .github/workflows/object_detection.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,89 @@
name: Object Detection

on:
pull_request:
branches:
- main
paths:
- ./recipes/computer_vision/object_detection/**
- .github/workflows/object_detection.yaml
push:
branches:
- main
paths:
- ./recipes/computer_vision/object_detection/**
- .github/workflows/object_detection.yaml

workflow_dispatch:

env:
REGISTRY: ghcr.io
REGISTRY_ORG: containers
RECIPE_NAME: object_detection
RECIPE_TYPE: computer_vision
IMAGE_NAME: object_detection_client

jobs:
chatbot-build-and-push:
if: "!contains(github.event.pull_request.labels.*.name, 'hold-tests')"
runs-on: ubuntu-22.04
permissions:
contents: read
packages: write
services:
registry:
image: registry:2.8.3
ports:
- 5000:5000
steps:
- uses: actions/[email protected]

- name: Install qemu dependency
run: |
sudo apt-get update
sudo apt-get install -y qemu-user-static
- name: Build Image
id: build_image
uses: redhat-actions/[email protected]
with:
image: ${{ env.REGISTRY }}/${{ env.REGISTRY_ORG }}/${{ env.IMAGE_NAME }}
tags: latest
platforms: linux/amd64,linux/arm64
containerfiles: ./recipes/${{ env.RECIPE_TYPE }}/${{ env.RECIPE_NAME }}/app/Containerfile
context: recipes/${{ env.RECIPE_TYPE }}/${{ env.RECIPE_NAME }}/app

- name: Set up Python
uses: actions/[email protected]
with:
python-version: '3.11'

- name: Install Dependencies
working-directory: ./recipes/${{ env.RECIPE_TYPE }}/${{ env.RECIPE_NAME }}/app
run: make install

- name: Download model
working-directory: ./models
run: make download-model-facebook-detr-resnet-101

- name: Run Functional Tests
shell: bash
run: make functional-tests
working-directory: ./recipes/${{ env.RECIPE_TYPE }}/${{ env.RECIPE_NAME }}

- name: Login to Registry
if: github.event_name == 'push' && github.ref == 'refs/heads/main'
uses: redhat-actions/[email protected]
with:
registry: ${{ env.REGISTRY }}
username: ${{ github.actor }}
password: ${{ secrets.GITHUB_TOKEN }}

- name: Push Image
id: push_image
if: github.event_name == 'push' && github.ref == 'refs/heads/main'
uses: redhat-actions/[email protected]
with:
image: ${{ steps.build_image.outputs.image }}
tags: ${{ steps.build_image.outputs.tags }}
registry: ${{ env.REGISTRY }}
1 change: 1 addition & 0 deletions model_servers/object_detection_python/tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@
IMAGE_NAME = os.getenv("IMAGE_NAME", "containers/object_detection_python:latest")
MODEL_NAME = os.getenv("MODEL_NAME", "facebook/detr-resnet-101")
MODELS_DIR = os.getenv("MODELS_DIR", "/app/models")

MODEL_PATH = f"{MODELS_DIR}/{MODEL_NAME}"

PORT = os.getenv("PORT", 8000)
Expand Down
9 changes: 9 additions & 0 deletions recipes/computer_vision/object_detection/Makefile
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
SHELL := /bin/bash
APP ?= object_detection_client
PORT ?= 8501

include ../../common/Makefile.common

RECIPE_BINARIES_PATH := $(shell realpath ../../common/bin)
RELATIVE_MODELS_PATH := ../../../models
RELATIVE_TESTS_PATH := tests
7 changes: 1 addition & 6 deletions recipes/computer_vision/object_detection/tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,14 +13,9 @@
except:
PORT = 8000

if 'IMAGE' not in os.environ:
IMAGE = 'ghcr.io/containers/model_servers/object_detection_python:latest'
else:
IMAGE = os.environ['IMAGE']

IMAGE = os.getenv("IMAGE", "ghcr.io/containers/object_detection_python:latest")
MODEL_NAME=os.environ['MODEL_NAME']
MODEL_PATH=os.environ['MODEL_PATH']

BIND_MOUNT_OPTIONS = 'ro'

MS = pytest_container.Container(
Expand Down

0 comments on commit 9397da0

Please sign in to comment.