学习计算机视觉的过程中,会用到多张图片,很多时候图片的大小、比例各不相同在导入的时候挺麻烦的。所以就写了一个可以将一个文件夹里的图片批量处理成同样格式的代码。实现用的是OpenCV-Python。
import os
import cv2 as cv
from pylab import *
maxsize = (100, 100) # 定义缩略图的大小
path = r'E:\PycharmProjects\SIFT\pic\result\pic' # 目标文件夹的路径
# 读取整个文件夹的图片
def read_path(pathname):
imgname_list = os.listdir(pathname)
img_list = []
i = 0
# 图片列表
for imgname in imgname_list:
# 就算文件夹里有别的文件也没有瓜西!
# .endwith()可以把你想要的文件按照后缀名筛选出来
if imgname.endswith('.jpg'):
img = cv.imread(pathname + '/' + imgname)
img_n = cv.resize(img, maxsize, cv.INTER_AREA)
filename = path + str(i) + '.png'
cv.imwrite(filename, img_n) # need temporary files of the right size
i = i + 1
print(i)
return img_list
list = read_path(r'E:\PycharmProjects\SIFT\pic') # 图片文件夹的路径
print(list) # 输出文件名
对于个人还是很方便哒(゜∀。)