遍历图像目录并将它们全部旋转 x 度并保存到目录

2022-01-16 00:00:00 python opencv scipy image rotation

问题描述

我正在使用 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:Minicondaenvs.."
    path = "C:Minicondaenvsout.."
    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(path) 为您提供文件夹的内容列表(基本上只是名称),然后您需要使用 ndimage.imread()<打开这些文件/em> 函数.

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:Minicondaenvs.."
    path = "C:Minicondaenvsout.."

    # 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?

相关文章