mirror of
https://github.com/azaion/detections.git
synced 2026-06-21 13:31:07 +00:00
Roman Meshko
6ad4b700dd
ci/woodpecker/push/02-build-push Pipeline was successful
* Run tests * Run tests * Run tests * Run tests * Added rebuild * Added files for e2e tests * Added rebuild * Added rebuild * Added biuld TensorRT flag * Changed to use NumPy 1.x for Jetson * Make universal invocation * Make Cython constans * Changed to prepare onnx * Changed smoke-test to wait AI conversion * Added step for model conversion * Changed to not run step in parallel * Push model to docker registry * Push model to docker registry * Push model to docker registry
44 lines
1.6 KiB
Bash
44 lines
1.6 KiB
Bash
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
COMPOSE="${COMPOSE:-docker compose -f docker-compose.test.yml --profile jetson}"
|
|
REGISTRY_HOST="${REGISTRY_HOST:?REGISTRY_HOST is required}"
|
|
ENGINE_REPOSITORY="${JETSON_ENGINE_REPOSITORY:-$REGISTRY_HOST/azaion/detections-jetson-engine}"
|
|
BRANCH="${CI_COMMIT_BRANCH:-local}"
|
|
ENGINE_TAG="${JETSON_ENGINE_TAG:-$(printf '%s' "$BRANCH" | tr -c 'A-Za-z0-9_.-' '-')}"
|
|
OUT_DIR="${JETSON_ENGINE_OUT_DIR:-results/jetson-engine}"
|
|
|
|
mkdir -p "$OUT_DIR/models"
|
|
|
|
loader_id="$($COMPOSE ps -q mock-loader)"
|
|
if [[ -z "$loader_id" ]]; then
|
|
echo "ERROR: mock-loader container is not running"
|
|
exit 1
|
|
fi
|
|
|
|
docker cp "$loader_id:/models/models/." "$OUT_DIR/models/"
|
|
find "$OUT_DIR/models" -maxdepth 1 -type f ! -name 'azaion*.engine' -delete
|
|
|
|
engine_count="$(find "$OUT_DIR/models" -maxdepth 1 -type f -name 'azaion*.engine' | wc -l | tr -d ' ')"
|
|
if [[ "$engine_count" == "0" ]]; then
|
|
echo "ERROR: no converted TensorRT engine found in mock-loader /models/models"
|
|
find "$OUT_DIR/models" -maxdepth 2 -type f -print
|
|
exit 1
|
|
fi
|
|
|
|
echo "--- Converted TensorRT engine files:"
|
|
find "$OUT_DIR/models" -maxdepth 1 -type f -name 'azaion*.engine' -print -exec ls -lh {} \;
|
|
|
|
image="$ENGINE_REPOSITORY:$ENGINE_TAG"
|
|
echo "--- Building Jetson engine artifact image: $image"
|
|
docker build -f engine-artifact.Dockerfile -t "$image" "$OUT_DIR/models"
|
|
docker push "$image"
|
|
|
|
if [[ -n "${CI_COMMIT_SHA:-}" ]]; then
|
|
sha_tag="$(printf '%s' "$CI_COMMIT_SHA" | cut -c1-12)"
|
|
docker tag "$image" "$ENGINE_REPOSITORY:$sha_tag"
|
|
docker push "$ENGINE_REPOSITORY:$sha_tag"
|
|
fi
|
|
|
|
echo "--- Published Jetson engine artifact image: $image"
|