Python+Opencv对多个图片进行任意倍数下采样并保存到指定路径
使用Python代码+Opencv库对多张图片下采样并保存到指定路径
对单个图片进行下采样很容易,我们可以使用opencv的函数cv2.pyrDown(),那么多个图片该怎么做呢?经过我2个小时的代码编写(小白勿喷),终于总结了一套适用于任何图像,任何倍数,任何数量图像的批量下采样方法。
废话不多说,直接开始:
首先导入必要的库
import cv2
import os.path
然后我们读取文件夹里面的图像数量 并返回filenum,这里的路径到文件夹即可,注意路径中的分隔符不能为 \ ,(可以为 / 或 \ ),否则会报错
def countFile(dir):
# 输入文件夹
tmp = 0
for item in os.listdir(dir):
if os.path.isfile(os.path.join(dir, item)):
tmp += 1
else:
tmp += countFile(os.path.join(dir, item))
return tmp
filenum = countFile("D:\\model\\super-resolution\\.div2k\images\\DIV2K_train_LR_bicubic\\X4") # 返回的是图片的张数
print(filenum)
注意这里的index为0或者1(为0时后面的代码中if index<9,为1时if index <10,因为index决定了图片的初始索引),这个值也可以自己设置。n的值也很重要,决定了几倍下采样,在后面的代码中也有体现。
# filenum
n = 4
index = 1 #保存图片编号
num = 0 #处理图片计数
主程序:因为我的输入图片名为0001x4.png-0900x4.png,输出图片也是0001x4.png-0900x4.png,所以我得分情况讨论输入输出,各位程序员可以按自己得需求修改此代码,n的值可以决定是2倍下采样还是4倍下采样,程序员可以再添加函数修改,以进行多倍下采样。
for i in range(1,filenum+1):
########################################################
# 1.读取原始图片
if index < 10:
filename = "D:\\model\\super-resolution\\.div2k\images\\DIV2K_train_LR_bicubic\\X4\\000"+str(i)+".png"
elif index < 100:
filename = "D:\\model\\super-resolution\\.div2k\images\\DIV2K_train_LR_bicubic\\X4\\00"+str(i)+".png"
else:
filename = "D:\\model\\super-resolution\\.div2k\images\\DIV2K_train_LR_bicubic\\X4\\0"+str(i)+".png"
print(filename)
original_image = cv2.imread(filename)
# 2.下采样
if n == 2:
img_1 = cv2.pyrDown(original_image)
if n == 4:
img_1 = cv2.pyrDown(original_image)
img_1 = cv2.pyrDown(img_1)
#3.将下采样图片保存到指定路径当中
if index < 10:
cv2.imwrite("D:\\model\\super-resolution\\.div2k\\1\\000" + str(index) + ".png", img_1)
elif index < 100:
cv2.imwrite("D:\\model\\super-resolution\\.div2k\\1\\00" + str(index) + ".png", img_1)
else:
cv2.imwrite("D:\\model\\super-resolution\\.div2k\\1\\0" + str(index) + ".png", img_1)
num = num + 1
print("正在为第"+ str(num) + "图片采样......")
index = index + 1
结果展示
原图大小:
下采样以后:
程序的全部代码
import cv2
#import def_Gaussian as dg
#import time
import os.path
#import glob
#####################################################################################################################
#读取文件夹里面的图像数量 并返回filenum
def countFile(dir):
# 输入文件夹
tmp = 0
for item in os.listdir(dir):
if os.path.isfile(os.path.join(dir, item)):
tmp += 1
else:
tmp += countFile(os.path.join(dir, item))
return tmp
filenum = countFile("D:\\model\\super-resolution\\.div2k\images\\DIV2K_train_LR_bicubic\\X4") # 返回的是图片的张数
print(filenum)
# filenum
n = 4
index = 1 #保存图片编号
num = 0 #处理图片计数
for i in range(1,filenum+1):
########################################################
# 1.读取原始图片
if index < 10:
filename = "D:\\model\\super-resolution\\.div2k\images\\DIV2K_train_LR_bicubic\\X4\\000"+str(i)+".png"
elif index < 100:
filename = "D:\\model\\super-resolution\\.div2k\images\\DIV2K_train_LR_bicubic\\X4\\00"+str(i)+".png"
else:
filename = "D:\\model\\super-resolution\\.div2k\images\\DIV2K_train_LR_bicubic\\X4\\0"+str(i)+".png"
print(filename)
original_image = cv2.imread(filename)
# 2.下采样
if n == 2:
img_1 = cv2.pyrDown(original_image)
if n == 4:
img_1 = cv2.pyrDown(original_image)
img_1 = cv2.pyrDown(img_1)
#3.将下采样图片保存到指定路径当中
if index < 10:
cv2.imwrite("D:\\model\\super-resolution\\.div2k\\1\\000" + str(index) + ".png", img_1)
elif index < 100:
cv2.imwrite("D:\\model\\super-resolution\\.div2k\\1\\00" + str(index) + ".png", img_1)
else:
cv2.imwrite("D:\\model\\super-resolution\\.div2k\\1\\0" + str(index) + ".png", img_1)
num = num + 1
print("正在为第"+ str(num) + "图片采样......")
index = index + 1