Создание BAT-файла для цикла запуска скрипта Python через всю папку - PullRequest
0 голосов
/ 27 января 2019

Дело в том, что сейчас у меня есть скрипт Python для визуализации.Если я запускаю его в режиме CUDA, я всегда получаю сообщение об ошибке: RuntimeError: CUDA error: out of memory (как в режиме CUDA для визуализации отдельного изображения, так и для визуализации папки).Однако, когда я запускаю сценарий в режиме ЦП, мой ЦП останавливается, и для прохождения одной визуализации изображения требуется около 30 секунд.

Но когда я пытаюсь выполнить цикл for внутри скрипта python для запуска визуализации через весь набор данных (около 500 изображений), мой компьютер падает на 5-м изображении (я подозреваю, что, возможно, есть некоторые переменные, которые я не сделалпосле каждого цикла for, я потратил время на этот подход, не сработал).

Поэтому я решаю отказаться от изящности кода Python и понять, могу ли я написать BAT-файл, который запускает мой скрипт визуализации по одному изображению за раз, а затем перейти к другому изображению, пока он не пройдет по всему набору данных?

Прямо сейчас у меня есть команда bash, работающая с одним изображением:

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

Я совершенно новичок в шелл-коде, не знаю, как запустить все файлы .jpg в этом "./ data / sample_folder_0 / "каталог.Нужна помощь!

1 Ответ

0 голосов
/ 28 января 2019

К счастью, я сам решил эту проблему.Но все равно спасибо вам, ребята.Надеюсь, кто-то еще с такой же ситуацией, с которой я столкнулся, смог увидеть этот пост и сэкономить свое время!

@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
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...