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

mlyde

(~~刚来，话说帖子能用markdown编辑吗，没发现有。~~)

                               
登录/注册后可看大图

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

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

---

用法
缩放裁剪后的x、y像素值在代码开始部分更改即可。
默认只使用 cut 函数，使用 cut2 函数时需在代码第18-20行更改。
注意：
1.缩放裁剪后会覆盖原文件，需要的话，请在缩放裁剪前备份图片。
2.没有做文件夹验证，请确认输入正确的文件夹路径，并确保文件夹中只有图片。
3.多次缩放可能会使图片变得模糊，尤其是文字边缘。

---

完整代码

1. import cv2
   
2. import os
   
3. import numpy as np
   
4. 
   
5. # cut 裁剪后的 xy
   
6. target_x = 286
   
7. target_y = 203
   
8. 
   
9. def get_dir(dir):
   
10.         """ 遍历文件夹下所有的文件名 """
    
11. 
    
12.         list1 = os.listdir(dir)
    
13.         for l in list1:
    
14.                 l = f'{dir}/{l}'
    
15.                 if(os.path.isdir(l)):
    
16.                         get_dir(l)
    
17.                 else:
    
18.                         cut(l)
    
19.                         # cut2(l,x=800)
    
20.                         # cut2(l,y=400)
    
21. 
    
22. def cut(inputdir = './t.jpg'):
    
23.         """ 图片裁剪 """
    
24. 
    
25.         # 裁剪后的文件名为
    
26.         # outputdir = inputdir[:-4] + '_over.jpg'
    
27.         # 设置为相同文件名,覆盖原文件
    
28.         outputdir = inputdir
    
29. 
    
30.         # img = cv2.imread(inputdir)        # imread读取不能包含中文文件名
    
31.         img = cv2.imdecode(np.fromfile(inputdir, dtype=np.uint8), cv2.IMREAD_UNCHANGED)
    
32. 
    
33.     # imdecode读取图像默BGR通道排列,
    
34.     # img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)        # 转换为RGB
    
35. 
    
36.         # img.shape:[height,width,channel]
    
37.         in_y, in_x = img.shape[:2]
    
38.         print(img.shape)
    
39. 
    
40.         scale = target_x / target_y
    
41.         scale1 = in_x / in_y
    
42. 
    
43.         if(scale1 >= scale):
    
44.                 size = (int(in_x/(in_y/target_y)), target_y)
    
45.                 # print(1, size)
    
46.         elif(scale1 < scale):
    
47.                 size = (target_x, int(in_y/(in_x/target_x)))
    
48.                 # print(2, size)
    
49.         else:
    
50.                 size = (target_x, target_y)
    
51.                 print('error')
    
52. 
    
53.         # 缩放到size=(x,y)
    
54.         resized = cv2.resize(img, size, interpolation=cv2.INTER_AREA)
    
55. 
    
56.         # 展示裁剪后的图片
    
57.         # cv2.imshow('image', resized)
    
58.         # cv2.waitKey(0)
    
59.         # print('x', resized.shape[1], 'y', resized.shape[0])
    
60. 
    
61.         if(resized.shape[1] == target_x):
    
62.                 # x=target_x
    
63.                 y0 = resized.shape[0] // 2 - target_y//2
    
64.                 y1 = y0 + target_y
    
65.                 x0 = 0
    
66.                 x1 = target_x
    
67.         if(resized.shape[0] == target_y):
    
68.                 # y=target_y
    
69.                 y0 = 0
    
70.                 y1 = target_y
    
71.                 x0 = resized.shape[1] // 2 - target_x//2
    
72.                 x1 = x0 + target_x
    
73. 
    
74.         output_img = resized[y0:y1, x0:x1]
    
75. 
    
76.         # cv2.imwrite(outputdir, output_img)        # imwrite保存文件名不能包含中文
    
77.         cv2.imencode('.jpg', output_img)[1].tofile(outputdir)
    
78. 
    
79. def cut2(inputdir = './t.jpg', x=0, y=0):
    
80.         """ 等比缩放,不裁剪 """
    
81.         # outputdir = inputdir[:-4] + '_over.jpg'
    
82.         outputdir = inputdir
    
83.         img = cv2.imdecode(np.fromfile(inputdir, dtype=np.uint8), cv2.IMREAD_UNCHANGED)
    
84. 
    
85.         in_y, in_x = img.shape[:2]
    
86.         if(x):
    
87.                 # 等比缩小为x=
    
88.                 fxy = x/in_x
    
89.         elif(y):
    
90.                 # 等比缩小为y=
    
91.                 fxy = y/in_y
    
92.         else:
    
93.                 fxy = 1
    
94. 
    
95.         # 按比例缩放,fx:x轴缩放比例,fy:y轴缩放比例
    
96.         output_img = cv2.resize(img, (0,0), fx=fxy, fy=fxy, interpolation=cv2.INTER_AREA)
    
97.         cv2.imencode('.jpg', output_img)[1].tofile(outputdir)
    
98. 
    
99. if __name__ == "__main__":
    
100.         original_dir = input('文件夹路径:')
     
101.         get_dir(original_dir)

复制代码