Skip to content

Commit

Permalink
feat: add capi-yolov8 ensemble into smoke tests
Browse files Browse the repository at this point in the history
Signed-off-by: Jim Wang <[email protected]>
  • Loading branch information
jim-wang-intel committed Feb 14, 2024
1 parent dccd012 commit 9bcb413
Showing 1 changed file with 34 additions and 0 deletions.
34 changes: 34 additions & 0 deletions run_smoke_test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -280,6 +280,18 @@ waitForLogFile
verifyNonEmptyPipelineLog capi_yolov5_ensemble $input_src
teardown

#11. gst capi capi_yolov8_ensemble profile:
make build-capi_yolov8_ensemble
echo "Running capi_yolov8_ensemble profile..."
input_src="rtsp://127.0.0.1:8554/camera_0"
PIPELINE_PROFILE="capi_yolov8_ensemble" RENDER_MODE=0 sudo -E ./run.sh --platform core --inputsrc "$input_src"
status_code=$?
verifyStatusCode capi_yolov8_ensemble $status_code $input_src
# allowing some time to process
waitForLogFile
verifyNonEmptyPipelineLog capi_yolov8_ensemble $input_src
teardown

#-----------------------------------------------------------------------------------------------------------------
# tests for running on device GPU
#
Expand Down Expand Up @@ -317,6 +329,17 @@ waitForLogFile
verifyNonEmptyPipelineLog capi_yolov5_ensemble_gpu $input_src
teardown

# capi_yolov8 ensemble
echo "Running capi_yolov8_ensemble profile on GPU.0..."
input_src="rtsp://127.0.0.1:8554/camera_0"
PIPELINE_PROFILE="capi_yolov8_ensemble" DEVICE="GPU" RENDER_MODE=0 sudo -E ./run.sh --platform dgpu.0 --inputsrc "$input_src"
status_code=$?
verifyStatusCode capi_yolov8_ensemble_gpu $status_code $input_src
# allowing some time to process
waitForLogFile
verifyNonEmptyPipelineLog capi_yolov8_ensemble_gpu $input_src
teardown

source benchmark-scripts/get-gpu-info.sh
if [ "$HAS_ARC" != 1 ]
then
Expand Down Expand Up @@ -357,3 +380,14 @@ verifyStatusCode capi_yolov5_ensemble_ARC_gpu $status_code $input_src
waitForLogFile
verifyNonEmptyPipelineLog capi_yolov5_ensemble_ARC_gpu $input_src
teardown

# capi_yolov8 ensemble
echo "Running capi_yolov8_ensemble profile on GPU.1..."
input_src="rtsp://127.0.0.1:8554/camera_0"
PIPELINE_PROFILE="capi_yolov8_ensemble" DEVICE="GPU" RENDER_MODE=0 sudo -E ./run.sh --platform dgpu.1 --inputsrc "$input_src"
status_code=$?
verifyStatusCode capi_yolov8_ensemble_ARC_gpu $status_code $input_src
# allowing some time to process
waitForLogFile
verifyNonEmptyPipelineLog capi_yolov8_ensemble_ARC_gpu $input_src
teardown

0 comments on commit 9bcb413

Please sign in to comment.