Upload layer_diff_dataset/test.py with huggingface_hub

layer_diff_dataset/test.py

@@ -0,0 +1,14 @@
+import cv2
+import numpy as np
+rgba = cv2.imread('/mnt/workspace/workgroup/sihui.jsh/layer_diff_dataset/try/rgba/1#Accessories#1#Bag#2339506821_83cf9f1d22_o.png',cv2.IMREAD_UNCHANGED)
+mask = rgba[:,:,3]
+
+# 检查除了0和255以外的值是否存在
+# unique_values = np.unique(mask)
+other_values_exist = not np.all(np.isin(unique_values, [0, 255]))
+cv2.imwrite('mask.png',mask)
+# 输出结果
+if other_values_exist:
+    print("Mask contains values other than 0 and 255.")
+else:
+    print("Mask contains only 0 and 255.")