Python批量处理图片对比度并且保存

在别人帖子上进行了改进，加入了读取文件夹下的图片并进行批量处理并保存的模块

import cv2
import sys
import os
import numpy as np
from PIL import Image
import matplotlib.pyplot as plt
"""对比度调整算法：主要是对RGB空间进行调整。设定合适的RGB阈值，并在此阈值基础上计算出合适的调整系数进行对比度调整。参考CSDN博客：https://blog.csdn.net/maozefa/article/details/7069001
"""
def ContrastAlgorithm(rgb_img, contrast=0.5, threshold=0.5):img = rgb_img * 1.0img_out = img# 增量等于1，按灰度阈值最多调整成八种颜色：# 黑、红、绿、蓝、黄(255,255,0)、品红(255,0,255)、青(0,255,255)、白if contrast == 1:# newRGB = RGB >= Threshold? 255 : 0mask_1 = img >= threshold * 255.0rgb1 = 255.0rgb2 = 0img_out = rgb1 * mask_1 + rgb2 * (1 - mask_1)# 增量大于0小于1elif contrast >= 0:alpha = 1 - contrastalpha = 1 / alpha - 1img_out[:, :, 0] = img[:, :, 0] + (img[:, :, 0] - threshold * 255.0) * alphaimg_out[:, :, 1] = img[:, :, 1] + (img[:, :, 1] - threshold * 255.0) * alphaimg_out[:, :, 2] = img[:, :, 2] + (img[:, :, 2] - threshold * 255.0) * alpha# 增量小于0else:alpha = contrastimg_out[:, :, 0] = img[:, :, 0] + (img[:, :, 0] - threshold * 255.0) * alphaimg_out[:, :, 1] = img[:, :, 1] + (img[:, :, 1] - threshold * 255.0) * alphaimg_out[:, :, 2] = img[:, :, 2] + (img[:, :, 2] - threshold * 255.0) * alpha# img_out = img_out / 255.0return img_out
#批量修改对比度并保存
def Save2dir(data_dir,save_dir):if not os.path.exists(save_dir):os.mkdir(save_dir)for name in os.listdir(data_dir):save_name = name[:-4] + "hh.png"img = cv2.imread(os.path.join(data_dir,name))img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)img_new = ContrastAlgorithm(img, contrast, threshold)# plt.imshow(img_new)# plt.savefig(os.path.join(save_dir,save_name))img_new = img_new[:, :, ::-1]cv2.imwrite(os.path.join(save_dir,save_name), img_new)#  run : python Contrast.py (path) (contrast) (threshold)
def show():path = '/home/ljs/Desktop/test/Screenshot from bearing.mp4.png' #单张图片测试img = cv2.imread(path)img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)img_new = ContrastAlgorithm(img, contrast, threshold)plt.figure("img_original")plt.imshow(img / 255.0)plt.axis('off')plt.figure("img_contrast")plt.imshow(img_new / 255.0)plt.axis('off')plt.show()def save():Data_dir = "/home/ljs/Desktop/test"  # 要改变的图片的路径文件夹Save_dir = "/home/ljs/Desktop/test"  # 要保存的图片的路径文件夹Save2dir(Data_dir,Save_dir)if __name__ == '__main__':contrast = 0.5  # 范围：-1至1threshold = 0.5  # 范围：0至1show() # 查看一张图的效果# save() # 进行批量处理并且保存

Python批量处理图片对比度并且保存相关推荐

如何用python批量处理图片大小_python批量修改图片大小的方法
本文实例为大家分享了python批量修改图片大小的具体代码,供大家参考,具体内容如下引用的模块 from PIL import Image Image的使用 def resize_image(img ...
python 批量处理图片文件（做到图片不变形）
对此次项目中主要涉及到的内容做以下说明: 1.需要收集大量资料(包括收集不同信息和图片): 2.资料太多需要做同样的操作处理(包括修改文件名和修改图片尺寸). 一.修改文件名 import os os ...
python批量_Python学习第九十三天：Python批量处理图片
1.Pillow库介绍Pillow是Python里的图像处理库,提供了了广泛的文件格式支持,强大的图像处理能力,主要包括图像储存.图像显示.格式转换以及基本的图像处理操作等: 安装 pip insta ...
python批量处理图片属性_python PIL 批量处理处理图片
python PIL 批量处理处理图片客户给一堆图片要传到后台,图片太大了,上百张图用photoshop改太慢,就想到用python写个简单的批处理.功能简单就是把原图按比例缩小,代码更简单 20多 ...
python批量处理图片颜色反转_Python批量处理图片
1.Pillow库介绍 Pillow是Python里的图像处理库,提供了了广泛的文件格式支持,强大的图像处理能力,主要包括图像储存.图像显示.格式转换以及基本的图像处理操作等: 安装 pip inst ...
用python批量处理图片_Python学习九十三天：Python批量处理图片
1.Pillow库介绍 Pillow是Python里的图像处理库,提供了了广泛的文件格式支持,强大的图像处理能力,主要包括图像储存.图像显示.格式转换以及基本的图像处理操作等: 安装 pip inst ...
Python批量处理图片
1.Pillow库介绍 Pillow是Python里的图像处理库,提供了了广泛的文件格式支持,强大的图像处理能力,主要包括图像储存.图像显示.格式转换以及基本的图像处理操作等: 安装 pip inst ...
python 批量截取图片并保存原图片名
前言最近需要对图片进行批量截取处理,所以写下来作为笔记代码 (1)对每张图片都需要点选然后截图 import matplotlib.pyplot as plt import os from PIL ...
Python批量将Photoshop文件保存为图片
这是给zhenguo老师的投稿,已经收获50元稿费. 困扰我的问题我是一名电商美工设计师,俗称电商图片搬运工,我的桌面经常是这样的.再就是这样的.各种的psd文件再加上文字无法表达色彩和构图的关键信 ...

Python批量处理图片对比度并且保存

Python批量处理图片对比度并且保存相关推荐

最新文章

热门文章