addproject

zhyy/library/CommonFun/handle_picture.py

@@ -0,0 +1,56 @@
+# -*- coding:utf-8 -*-
+"""
+Author: qiaoxinjiu
+Email: qiaoxinjiu@sparkedu.com
+Create Date: 2022/05/08 5:58 下午
+"""
+import os
+from PIL import Image
+
+def image_gray(img):
+    # 打开图片
+    img = Image.open(img)
+
+    # 计算平均灰度值
+    gray_sum = 0
+    count = 0
+    for x in range(img.width):
+        for y in range(img.height):
+            if img.mode == "RGB":
+                r, g, b = img.getpixel((x, y))
+                gray_sum += (r + g + b) / 3
+            elif img.mode == "L":
+                gray_value = img.getpixel((x, y))
+                gray_sum += gray_value
+            count += 1
+
+    avg_gray = gray_sum / count
+    return avg_gray
+
+
+def find_image(folder_path):
+    # 定义一个列表存储图片路径
+    images = []
+    # 遍历文件夹下的所有文件
+    for root, dirs, files in os.walk(folder_path):
+        for file in files:
+            file_path = os.path.join(root, file)
+            # 处理每个文件,将其添加到列表中
+            images.append(file_path)
+    return images
+
+
+def assert_run(folder_path):
+    images = find_image(folder_path)
+    for img in images:
+        gray = image_gray(img)
+        # 灰度值小于50，将认为是黑图
+        if gray < 50:
+            print(img, "：", gray)
+
+
+if __name__ == "__main__":
+    # image_gray()
+    # find_image()
+    folder_path = r'D:\picture'
+    assert_run(folder_path)