Python+Opencv对多个图片批量进行任意倍数下采样并保存到指定路径

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