问题是现在我有一个可视化的python脚本。如果我在CUDA模式下运行它,我总是得到错误信息:RuntimeError: CUDA error: out of memory(在CUDA模式下,用于单图像可视化和文件夹可视化)。然而,当我在CPU模式下运行脚本时,我的CPU会停止工作,并且需要大约30秒来运行一个图像可视化。
但是,当我尝试在python脚本中使用for循环来运行整个数据集(大约500张图像)的可视化时,我的计算机在第5张图像处崩溃了(我怀疑可能有一些变量没有在每个for循环之后释放,我花了一些时间在这种方法上,没有很好地工作)。
因此,我决定放弃python代码的优雅,并设想如果我可以编写一个BAT文件,一次一个图像地运行我的可视化脚本,然后转到另一个图像,直到它遍历整个数据集?
现在,我已经将bash命令用于单个镜像:
@echo off
python main_single_img.py -i "./data/sample_folder_0/image0.jpg" --no-cuda %*
pause我对shell代码完全陌生,不知道如何运行这个“./ .jpg /sample_folder_0/”目录中的所有数据文件。需要帮助!
发布于 2019-01-28 06:02:27
幸运的是,我自己刚刚解决了这个问题。不过,还是要谢谢你们。希望其他和我遇到同样情况的人可以看到这篇文章,节省他们的时间!
@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 %*
)
pausehttps://stackoverflow.com/questions/54392559
复制相似问题