Upload folder using huggingface_hub

ct/create_lists.py

@@ -0,0 +1,32 @@
+import random
+import os
+
+exs = []
+for f in os.listdir("imgs"):
+    ex = f.split(".jpg")[0]
+    exs.append(ex)
+
+random.shuffle(exs)
+
+fp = open("train.csv")
+
+train_exs = []
+
+for line in fp:
+    train_exs.append(line.strip().split(",")[0])
+fp.close()
+
+test_exs = [ex for ex in exs if ex not in train_exs]
+print(len(exs), len(test_exs))
+#train_exs = exs[:3896]
+#test_exs = exs[int(len(exs)*0.2) + 1:]
+
+#train_fp = open("train.csv", "w")
+#for ex in train_exs:
+#    train_fp.write(ex + ",1\n")
+#train_fp.close()
+
+test_fp = open("test.csv", "w")
+for ex in test_exs:
+    test_fp.write(ex + ",1\n")
+test_fp.close()

ct/create_masks.py

@@ -0,0 +1,26 @@
+from PIL import Image
+import cv2
+import os
+import numpy as np
+
+i = 0
+threshold = 240
+m = 0
+for img in os.listdir("orig_masks"):
+    mask = cv2.imread("orig_masks/" + img)
+    for d in range(3):
+        mask[:, :, d][mask[:, :, d] < threshold] = 0
+        mask[:, :, d][mask[:, :, d] > 0] = 1
+    mask = mask[:, :, 0] | mask[:, :, 1] | mask[:, :, 2]
+    mask = (mask * 255).astype(np.uint8)
+    #nonzero = []
+    #for i in mask[np.nonzero(mask)]:
+    #    nonzero.append(i)
+    if np.all(mask == 0):
+        print("The mask contains all zeros.")
+        continue
+    cv2.imwrite("masks/" + img, mask)
+    img_split = img.split("_")
+    os.system("cp orig_imgs/" + img_split[0] + "_" + img_split[2] + " imgs/")
+    i += 1
+    print("Done ", str(i), str(m))

ct/final.zip

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:87b03361d7b4d154fb15ee0887c83b12e9441b81158201916bcf1fdd529e355c
+size 488050930