功能介绍:从一个文件夹中读取图片,获得图片的像素最大值,并记录在txt文件中保存,同时应保存对应的文件名。
特别说明:图片文件为png格式,8bit的单层图(即灰度图),不确定此代码是否适用于其他文件类型,未做测试。
import numpy as np
import os
from PIL import Image
def get_path_list(file_dir, fname):
L = []
for root, dirs, files in os.walk(file_dir):
for file in sorted(files): # 遍历文件目录下每一个文件
if fname in file: # 判断是否包含指定字符串
L.append(os.path.join(root, file))
if __name__ == '__main__':
dir = './sparse' # 文件夹名
dirs = os.listdir(dir)
sparse_data_file = os.path.join('./depth_errors.txt')
for dir1 in dirs:
imgpath = dir + '/' +dir1
image = Image.open(imgpath)
image_max = np.max(image)
image_min = np.min(image)
message = 'image: %s, max: %s\n' % (dir1, image_max)
with open(sparse_data_file, "a") as file:
file.write('%s' % message)
print(message)