Batch Script for Replacing Parameters in infer.py File Based on User Selection

@echo off

setlocal enabledelayedexpansion

set index=1 echo 选择checkpoints文件夹下的文件夹：

for /f %%i in ('dir /b/ad checkpoints') do ( echo !index!. %%i set /a index+=1 set name[!index!]=%%i )

set /p select=请输入序号:

set project_name=!name[%select%]!

echo 选择文件夹'%project_name%'下的ckpt文件：

set index=1 for /f %%i in ('dir /b/ad checkpoints%project_name%*.ckpt') do ( echo !index!. %%i set /a index+=1 set name[!index!]=%%i )

set /p select=请输入序号:

set ckpt_name=!name[%select%]!

echo 替换infer.py文件中的参数

set "file=infer.py" set "search=project_name = 'demo'" set "replace=project_name = '%project_name%'" (for /f "delims=" %%i in (%file%) do ( set "line=%%i" setlocal enabledelayedexpansion if "!line!"=="%search%" (echo(%replace%) else echo(!line!) endlocal ))>"%file%.tmp" move /y "%file%.tmp" "%file%"

set "file=infer.py" set "search=model_path = f'./checkpoints/{project_name}/model_ckpt_steps_246000.ckpt'" set "replace=model_path = f'./checkpoints/%project_name%/%ckpt_name%'" (for /f "delims=" %%i in (%file%) do ( set "line=%%i" setlocal enabledelayedexpansion if "!line!"=="%search%" (echo(%replace%) else echo(!line!) endlocal ))>"%file%.tmp" move /y "%file%.tmp" "%file%"

echo 替换完成