创建一个BAT文件以循环通过整个文件夹运行python脚本

Question

The thing is right now I have a visualization python script. If I run it on CUDA mode, I always get the error message of: RuntimeError: CUDA error: out of memory (on both CUDA mode for single image visualization and folder visualization). However, when I run the script on CPU mode, my CPU stands and it takes like 30seconds to run through one image visualization.

But when I try to have a for loop inside the python script to run the visualization through a whole dataset (about 500 images), my computer crashed at the 5th image(I suspect maybe there are some variables I didn't release after each for loop, I spent time on this approach, didn't work well).

Therefore I decide to discard the gracefulness of python code and figure if I could write a BAT file that runs my visualization script one image at a time and then go to another image till it loops through the whole dataset?

Right now I have the bash command works for a single image:

@echo off
python main_single_img.py -i "./data/sample_folder_0/image0.jpg" --no-cuda %*
pause

I'm completely new to shell code, no idea how to run through all .jpg files in this "./data/sample_folder_0/" directory. Need help!

Answer 1

Fortunately, I just solved this problem myself. But thank you guys anyway. Hope someone else with the same situation I encountered could see this post and save their time!

@echo off
for %%f in (./data/sample_folder_0/*.jpg) do (
    echo %%~nf
    python main_single_img.py -i ./data/sample_folder_0/%%f --no-cuda %*
)
pause