循环遍历图像目录并将它们全部旋转x度并保存到目录 [英] Loop through directory of images and rotate them all x degrees and save to directory

问题描述

我正在使用Python，Open，Numpy和Scipy。我有一个想要按特定角度旋转的图像目录。我想编写这个脚本。我正在使用它， OpenCV Python旋转在特定点周围X度的图像但它似乎没有像我想象的那样完全管道。我得到了一个无效的轮换计划，但我不认为我应该得到这个。

I am using Python,Open, Numpy, and Scipy. I have a directory of images that I want to rotate by certain angles. I want to script this. I am using this, OpenCV Python rotate image by X degrees around specific point but it doesn't seem to pipeline exactly as I envisioned. I get an invalid rotation plan specified, but I don't think that I should be getting this.

以下是我的代码：

from scipy import ndimage
import numpy as np
import os
import cv2

def main():
    outPath = "C:\Miniconda\envs\.."
    path = "C:\Miniconda\envs\out\.."
    for image_to_rotate in os.listdir(path):
        rotated = ndimage.rotate(image_to_rotate, 45)
        fullpath = os.path.join(outPath, rotated)

  if __name__ == '__main__':
     main()

推荐答案

在旋转之前，您需要实际读取图像文件。你当前的代码正在做的只是迭代文件（和目录）的名称。

You need to actually read the image file before rotating them. What your current code is doing is just iterating through the names of the files (and directories).

os.listdir（路径）给你该文件夹的内容列表（基本上只是名称），然后您需要使用 ndimage.imread（）函数打开这些文件。

os.listdir(path) gives you a list of contents of the folder (basically just the name) and then you need to open these files using the ndimage.imread() function.

这应该有效：

from scipy import ndimage, misc
import numpy as np
import os
import cv2

def main():
    outPath = "C:\Miniconda\envs\.."
    path = "C:\Miniconda\envs\out\.."

    # iterate through the names of contents of the folder
    for image_path in os.listdir(path):

        # create the full input path and read the file
        input_path = os.path.join(path, image_path)
        image_to_rotate = ndimage.imread(input_path)

        # rotate the image
        rotated = ndimage.rotate(image_to_rotate, 45)

        # create full output path, 'example.jpg' 
        # becomes 'rotate_example.jpg', save the file to disk
        fullpath = os.path.join(outPath, 'rotated_'+image_path)
        misc.imsave(fullpath, rotated)

if __name__ == '__main__':
    main()

PS：这种迭代文件夹内容的方式只有在目录中只有文件且没有子目录时才有效。 os.listdir（path）将返回任何文件和子目录的名称。

PS: This way of iterating through the contents of the folder only works if there are only files in the directory and no sub-directories. os.listdir(path) will return the names of any files as well as sub-directories.

您可以学习如何仅列出文件在这篇文章的目录中：如何列出所有文件一个目录？

You can learn how to list only files in a directory from this post: How to list all files of a directory?

这篇关于循环遍历图像目录并将它们全部旋转x度并保存到目录的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！