File size: 778 Bytes
8dc9718
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
@echo off
setlocal enabledelayedexpansion

REM 设置默认源图像路径
set "default_src_image=assets\examples\source\s12.jpg"
set "src_image=%default_src_image%"
set "animal_param="
set "paste_back="

REM 解析命名参数
:parse_args
if "%~1"=="" goto end_parse_args
if /i "%~1"=="--src_image" (
    set "src_image=%~2"
    shift
) else if /i "%~1"=="--animal" (
    set "animal_param=--animal"
) else if /i "%~1"=="--paste_back" (
    set "paste_back=--paste_back"
)
shift
goto parse_args
:end_parse_args

echo source image: [!src_image!]
echo use animal: [!animal_param!]
echo paste_back: [!paste_back!]

REM 执行Python命令
.\venv\python.exe .\run.py --cfg configs/trt_infer.yaml --realtime --dri_video 0 --src_image !src_image! !animal_param! !paste_back!

endlocal