Upload layer_diff_dataset/cv2_inpaint copy.py with huggingface_hub

layer_diff_dataset/cv2_inpaint copy.py

@@ -0,0 +1,31 @@
+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'
+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()