diff --git a/samples/gstreamer/gst_launch/detection_with_yolo/yolo_detect.sh b/samples/gstreamer/gst_launch/detection_with_yolo/yolo_detect.sh index efe25f2c..3b15a6ff 100755 --- a/samples/gstreamer/gst_launch/detection_with_yolo/yolo_detect.sh +++ b/samples/gstreamer/gst_launch/detection_with_yolo/yolo_detect.sh @@ -23,7 +23,7 @@ if [ "$1" = "--help" ] || [ "$1" = "-h" ]; then echo "" echo "Arguments:" echo " MODEL - Model name (default: yolox_s)" - echo " Supported: yolo_all, yolox-tiny, yolox_s, yolov7, yolov8s, yolov8n-obb, yolov8n-seg, yolov9c, yolov10s, yolo11s, yolo11s-obb, yolo11s-seg, yolo11s-pose" + echo " Supported: yolo_all, yolox-tiny, yolox_s, yolov7, yolov8s, yolov8n-obb, yolov8n-seg, yolov9c, yolov10s, yolo11s, yolo11s-obb, yolo11s-seg, yolo11s-pose, yolo26n, yolo26s, yolo26m, yolo26l, yolo26x, yolo26s-obb, yolo26s-seg, yolo26s-pose" echo " DEVICE - Device (default: GPU). Supported: CPU, GPU, NPU" echo " INPUT - Input source (default: Pexels video URL)" echo " OUTPUT - Output type (default: file). Supported: file, display, fps, json, display-and-json" @@ -33,7 +33,7 @@ if [ "$1" = "--help" ] || [ "$1" = "-h" ]; then exit 0 fi -MODEL=${1:-"yolox_s"} # Supported values: yolo_all, yolox-tiny, yolox_s, yolov7, yolov8s, yolov8n-obb, yolov8n-seg, yolov9c, yolov10s, yolo11s, yolo11s-obb, yolo11s-seg, yolo11s-pose +MODEL=${1:-"yolox_s"} # Supported values: yolo_all, yolox-tiny, yolox_s, yolov7, yolov8s, yolov8n-obb, yolov8n-seg, yolov9c, yolov10s, yolo11s, yolo11s-obb, yolo11s-seg, yolo11s-pose, yolo26n, yolo26s, yolo26m, yolo26l, yolo26x, yolo26s-obb, yolo26s-seg, yolo26s-pose DEVICE=${2:-"GPU"} # Supported values: CPU, GPU, NPU INPUT=${3:-"https://videos.pexels.com/video-files/1192116/1192116-sd_640_360_30fps.mp4"} OUTPUT=${4:-"file"} # Supported values: file, display, fps, json, display-and-json diff --git a/samples/windows/gstreamer/gst_launch/detection_with_yolo/yolo_detect.bat b/samples/windows/gstreamer/gst_launch/detection_with_yolo/yolo_detect.bat index 6d010232..71489e1d 100644 --- a/samples/windows/gstreamer/gst_launch/detection_with_yolo/yolo_detect.bat +++ b/samples/windows/gstreamer/gst_launch/detection_with_yolo/yolo_detect.bat @@ -17,6 +17,11 @@ if NOT DEFINED MODELS_PATH ( ) echo MODELS_PATH: %MODELS_PATH% +set MODELS_LIST=yolox-tiny yolox_s yolov5s yolov5su yolov7 yolov8s ^ +yolov8n-obb yolov8n-seg yolov9c yolov10s yolo11s ^ +yolo11s-seg yolo11s-obb yolo11s-pose yolo26n yolo26s ^ +yolo26m yolo26l yolo26x yolo26s-obb yolo26s-seg yolo26s-pose + @REM Parse arguments set MODEL=%1 if [%MODEL%]==[] set MODEL=yolox_s @@ -46,7 +51,7 @@ echo Usage: yolo_detect.bat [MODEL] [DEVICE] [INPUT] [OUTPUT] [PPBKEND] [PRECISI echo. echo Arguments: echo MODEL - Model name (default: yolox_s) -echo Supported: yolox-tiny, yolox_s, yolov5s, yolov5su, yolov7, yolov8s, yolov8n-obb, yolov8n-seg, yolov9c, yolov10s, yolo11s, yolo11s-obb, yolo11s-seg, yolo11s-pose +echo Supported: %MODELS_LIST% echo DEVICE - Device (default: GPU). Supported: CPU, GPU, NPU echo INPUT - Input source (default: Pexels video URL) echo OUTPUT - Output type (default: display). Supported: file, display, fps, json, display-and-json @@ -59,9 +64,11 @@ EXIT /B 0 @REM Validate model set VALID_MODEL=0 -for %%m in (yolox-tiny yolox_s yolov5s yolov5su yolov7 yolov8s yolov8n-obb yolov8n-seg yolov9c yolov10s yolo11s yolo11s-seg yolo11s-obb yolo11s-pose) do ( + +for %%m in (%MODELS_LIST%) do ( if "%MODEL%"=="%%m" set VALID_MODEL=1 ) + if %VALID_MODEL%==0 ( echo [91mERROR: Unsupported model: %MODEL%[0m EXIT /B 1