python 缩放并裁剪图片 制作图片缩略图

说明

现在有一文件夹中存在许多分辨率不同的图片或文件夹,需要裁剪至指定大小以便作为网页中的图片缩略图。

cut 函数,将图片裁剪为指定大小,统一分辨率,缩放后取图片中间部分,超出的部分直接裁剪掉。
还有一个函数 cut2,为等比缩放至x或y为定值。

用法

缩放裁剪后的x、y像素值在代码开始部分更改即可。
默认只使用 cut 函数,使用 cut2 函数时需在代码第18-20行更改。
注意:

1.缩放裁剪后会覆盖原文件,需要的话,请在缩放裁剪前备份图片。
2.没有做文件夹验证,请确认输入正确的文件夹路径,并确保文件夹中只有图片。
3.多次缩放可能会使图片变得模糊,尤其是文字边缘。

完整代码

import cv2
import os
import numpy as np

# cut 裁剪后的 xy
target_x = 286
target_y = 203

def get_dir(dir):
	""" 遍历文件夹下所有的文件名 """

	list1 = os.listdir(dir)
	for l in list1:
		l = f'{dir}/{l}'
		if(os.path.isdir(l)):
			get_dir(l)
		else:
			cut(l)
			# cut2(l,x=800)
			# cut2(l,y=400)

def cut(inputdir = './t.jpg'):
	""" 图片裁剪 """

	# 裁剪后的文件名为
	# outputdir = inputdir[:-4] + '_over.jpg'
	# 设置为相同文件名,覆盖原文件
	outputdir = inputdir

	# img = cv2.imread(inputdir)	# imread读取不能包含中文文件名
	img = cv2.imdecode(np.fromfile(inputdir, dtype=np.uint8), cv2.IMREAD_UNCHANGED)

    # imdecode读取图像默BGR通道排列, 
    # img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)	# 转换为RGB

	# img.shape:[height,width,channel]
	in_y, in_x = img.shape[:2]
	print(img.shape)

	scale = target_x / target_y
	scale1 = in_x / in_y

	if(scale1 >= scale):
		size = (int(in_x/(in_y/target_y)), target_y)
		# print(1, size)
	elif(scale1 < scale):
		size = (target_x, int(in_y/(in_x/target_x)))
		# print(2, size)
	else:
		size = (target_x, target_y)
		print('error')

	# 缩放到size=(x,y)
	resized = cv2.resize(img, size, interpolation=cv2.INTER_AREA)

	# 展示裁剪后的图片
	# cv2.imshow('image', resized)
	# cv2.waitKey(0)
	# print('x', resized.shape[1], 'y', resized.shape[0])

	if(resized.shape[1] == target_x):
		# x=target_x
		y0 = resized.shape[0] // 2 - target_y//2
		y1 = y0 + target_y
		x0 = 0
		x1 = target_x
	if(resized.shape[0] == target_y):
		# y=target_y
		y0 = 0
		y1 = target_y
		x0 = resized.shape[1] // 2 - target_x//2
		x1 = x0 + target_x

	output_img = resized[y0:y1, x0:x1]

	# cv2.imwrite(outputdir, output_img)	# imwrite保存文件名不能包含中文
	cv2.imencode('.jpg', output_img)[1].tofile(outputdir)

def cut2(inputdir = './t.jpg', x=0, y=0):
	""" 等比缩放,不裁剪 """
	# outputdir = inputdir[:-4] + '_over.jpg'
	outputdir = inputdir
	img = cv2.imdecode(np.fromfile(inputdir, dtype=np.uint8), cv2.IMREAD_UNCHANGED)

	in_y, in_x = img.shape[:2]
	if(x):
		# 等比缩小为x=
		fxy = x/in_x
	elif(y):
		# 等比缩小为y=
		fxy = y/in_y
	else:
		fxy = 1

	# 按比例缩放,fx:x轴缩放比例,fy:y轴缩放比例
	output_img = cv2.resize(img, (0,0), fx=fxy, fy=fxy, interpolation=cv2.INTER_AREA)
	cv2.imencode('.jpg', output_img)[1].tofile(outputdir)

if __name__ == "__main__":
	original_dir = input('文件夹路径:')
	get_dir(original_dir)

你可能感兴趣的:(python,opencv)