python - 如何加载、深入梦想和覆盖文件夹中的所有图像？

Question

到目前为止，我已经能够通过替换“图片”来“手动”处理图像

photo = ''directory/picture.jpg''

对于我正在处理的每张图像。

这是有效的，当然，但它很慢。

有任何想法吗？

我正在使用的代码：

from deepdreamer import model, load_image, recursive_optimize
import numpy as np
import PIL.Image
layer_tensor = ...

photo = "directory/picture.jpg"
img_result = load_image(filename='{}'.format(photo))

img_result = recursive_optimize(...)
img_result = np.clip(img_result, 0.0, 255.0)
img_result = img_result.astype(np.uint8)
result = PIL.Image.fromarray(img_result, mode='RGB')
result.save(photo)

Answer 1

假设所有图像文件都在同一个文件夹/目录中，您可以：

1. 将您的图像处理封装到一个函数中。
2. 使用 os.listdir() 查找所有文件名。
3. 遍历文件名，将每个文件名传递给 imageProcessing() 函数，该函数对每个图像执行操作。

蟒蛇代码：

from deepdreamer import model, load_image, recursive_optimize
import numpy as np
import PIL.Image
import os //for file management stuff

layer_tensor = ...

def imageProcess(aFile):
  photo = "directory/{aFile}"
  img_result = load_image(filename='{}'.format(photo))

  img_result = recursive_optimize(...)
  img_result = np.clip(img_result, 0.0, 255.0)
  img_result = img_result.astype(np.uint8)
  result = PIL.Image.fromarray(img_result, mode='RGB')
  result.save(photo)

for filename in os.listdir('dirname'):
     imageProcess(filename)

会是这样的。让我知道它是如何工作的，我没有运行代码。