initial commit: add image layering

debug.py

@@ -0,0 +1,37 @@
+from PIL import Image
+import numpy as np
+
+
+
+def alpha_cutout(img, threshold=100, dist=5):
+    arr = np.array(np.asarray(img))  # 获取图像数据，使用了numpy
+    r, g, b, a = np.rollaxis(arr, axis=-1)
+
+    mask = ((r > threshold)
+            & (g > threshold)
+            & (b > threshold)
+            & (np.abs(r - g) < dist)  # 将接近白色背景的也替换掉
+            & (np.abs(r - b) < dist)
+            & (np.abs(g - b) < dist)
+            )
+    arr[mask, 3] = 0
+
+    img = Image.fromarray(arr, mode='RGBA')  # 转换为图像格式
+    return img
+
+def cutout_csdn():
+
+    threshold = 100
+    dist = 5
+    img = Image.open("img.png").convert('RGBA')  # 增加Alpha通道
+
+    img = alpha_cutout(img, threshold, dist)
+
+
+    img.show()
+
+
+if __name__ == '__main__':
+    cutout_csdn()
+    # image = "img.png"
+    # do_cutout(image)