Upload layer_diff_dataset/cv2_inpaint copy 2.py with huggingface_hub

layer_diff_dataset/cv2_inpaint copy 2.py

@@ -0,0 +1,33 @@
+import numpy as np
+# from matplotlib import pyplot as plt
+import cv2
+import os
+
+jpeg_folder = '/mnt/workspace/workgroup/sihui.jsh/layer_diff_dataset/YoutubeVOS/JPEGImages'
+# vid_name = '0a8c467cc3'
+vid_name_list = os.listdir(jpeg_folder)
+for idx,vid_name in enumerate(vid_name_list):
+    img_path = os.path.join(jpeg_folder,vid_name,'00000.jpg')
+    mask_folder = '/mnt/workspace/workgroup/sihui.jsh/layer_diff_dataset/YoutubeVOS/mask_dilate'
+    mask_path = os.path.join(mask_folder,vid_name,'00000.png')
+    img = cv2.imread(img_path)     # input
+    mask = cv2.imread(mask_path,0)  # mask
+
+    dst_TELEA = cv2.inpaint(img,mask,3,cv2.INPAINT_TELEA)
+    cv2.imwrite(f'cv2_inpaint_single_{vid_name}_dilate.png',dst_TELEA)
+
+# dst_NS = cv2.inpaint(img,mask,3,cv2.INPAINT_NS)
+
+# plt.subplot(221), plt.imshow(img)
+# plt.title('degraded image')
+# plt.subplot(222), plt.imshow(mask, 'gray')
+# plt.title('mask image')
+# plt.subplot(223), plt.imshow(dst_TELEA)
+# plt.title('TELEA')
+# plt.subplot(224), plt.imshow(dst_NS)
+# plt.title('NS')
+
+# plt.tight_layout()
+# save the image
+# plt.savefig(f'cv2_inpaint_{vid_name}.png')
+# plt.show()