Upload layer_diff_dataset/cv2_inpaint.py with huggingface_hub

layer_diff_dataset/cv2_inpaint.py

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