Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand All @@ -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
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand All @@ -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
Expand Down
Loading