我创建了一个简单的 Python 程序来翻转多个图像文件并将其保存到另一个文件夹。我遇到的问题是文件在不考虑文件编号的情况下加载到数组中,因此输出文件与原始文件的命名不同(不包括 0.jpg 和 1.jpg)。
下面是我用来翻转图像的代码。
# import modules
from tqdm import tqdm
import numpy as np
import glob
import cv2
# directory path to images
training_images = glob.glob("original_images/*.jpg")
# load the images into an array
image_array = np.array([cv2.imread(file) for file in training_images])
# find how many contents exist inside the array
array_length = np.size(image_array)
# flip the images and save to flipped_images/ folder
for x in range(array_length):
number = str(x)
flipped = np.flip(image_array[x], 1)
cv2.imwrite('flipped_images/' + number + '.jpg', flipped)
有什么办法可以让我按照文件名顺序加载数据?提前致谢。
为什么不从输入名称创建输出名称?
# import modules
from tqdm import tqdm
import numpy as np
import glob
import cv2
# directory path to images
training_images = glob.glob("original_images/*.jpg")
for filename in training_images:
image = np.array(cv2.imread(filename))
flipped = np.flip(image, 1)
out_path = filename.replace("original_images","flipped_images")
cv2.imwrite(out_path, flipped)
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句