mirror of
https://github.com/azaion/detections.git
synced 2026-06-21 06:51:07 +00:00
Fixed pipeline
This commit is contained in:
@@ -34,7 +34,11 @@ steps:
|
|||||||
echo "ERROR: fixtures/models/azaion.onnx is missing; cannot convert Jetson engine"
|
echo "ERROR: fixtures/models/azaion.onnx is missing; cannot convert Jetson engine"
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
ls -lh fixtures/image_small.jpg fixtures/models/azaion.onnx
|
if [ ! -f fixtures/models/azaion.int8_calib.cache ]; then
|
||||||
|
echo "ERROR: fixtures/models/azaion.int8_calib.cache is missing; cannot verify INT8 Jetson conversion"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
ls -lh fixtures/image_small.jpg fixtures/models/azaion.onnx fixtures/models/azaion.int8_calib.cache
|
||||||
|
|
||||||
E2E_PROFILE=jetson \
|
E2E_PROFILE=jetson \
|
||||||
E2E_WAIT_FOR_ENGINE_ENABLED=1 \
|
E2E_WAIT_FOR_ENGINE_ENABLED=1 \
|
||||||
@@ -48,6 +52,6 @@ steps:
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
bash scripts/publish_jetson_engine.sh
|
REQUIRE_INT8_ENGINE=1 bash scripts/publish_jetson_engine.sh
|
||||||
volumes:
|
volumes:
|
||||||
- /var/run/docker.sock:/var/run/docker.sock
|
- /var/run/docker.sock:/var/run/docker.sock
|
||||||
|
|||||||
@@ -65,7 +65,7 @@ services:
|
|||||||
- jetson
|
- jetson
|
||||||
build:
|
build:
|
||||||
context: ..
|
context: ..
|
||||||
dockerfile: Dockerfile.jetson
|
dockerfile: ${JETSON_DOCKERFILE:-Dockerfile.jetson}
|
||||||
runtime: nvidia
|
runtime: nvidia
|
||||||
depends_on:
|
depends_on:
|
||||||
- mock-loader
|
- mock-loader
|
||||||
|
|||||||
@@ -28,6 +28,15 @@ if [[ "$engine_count" == "0" ]]; then
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
if [[ "${REQUIRE_INT8_ENGINE:-0}" == "1" ]]; then
|
||||||
|
int8_engine_count="$(find "$OUT_DIR/models" -maxdepth 1 -type f -name 'azaion*.int8.engine' | wc -l | tr -d ' ')"
|
||||||
|
if [[ "$int8_engine_count" == "0" ]]; then
|
||||||
|
echo "ERROR: INT8 engine is required, but no azaion*.int8.engine file was produced"
|
||||||
|
find "$OUT_DIR/models" -maxdepth 1 -type f -name 'azaion*.engine' -print -exec ls -lh {} \;
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
echo "--- Converted TensorRT engine files:"
|
echo "--- Converted TensorRT engine files:"
|
||||||
find "$OUT_DIR/models" -maxdepth 1 -type f -name 'azaion*.engine' -print -exec ls -lh {} \;
|
find "$OUT_DIR/models" -maxdepth 1 -type f -name 'azaion*.engine' -print -exec ls -lh {} \;
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user