更改

tools/getpairs.py

@@ -188,7 +188,7 @@ class PairGenerator:
 
 
 if __name__ == "__main__":
-    original_path = '/home/lc/data_center/contrast_learning/contrast_test_data/test'
+    original_path = '/home/lc/data_center/scatter/val_extar'
     parent_dir = str(Path(original_path).parent)
     generator = PairGenerator()
 

tools/image_joint.py

@@ -0,0 +1,33 @@
+from PIL import Image, ImageDraw, ImageFont
+import os
+
+
+def merge_imgs(img1_path, img2_path, save_path, similar=None, label=None):
+        position = (50, 50)  # 文字的左上角坐标
+        color = (255, 0, 0)  # 红色文字，格式为 RGB
+        if not os.path.exists(os.sep.join([save_path, str(label)])):
+            os.makedirs(os.sep.join([save_path, str(label)]))
+        save_path = os.sep.join([save_path, str(label)])
+        img_name = os.path.basename(img1_path).split('.')[0]+'_'+os.path.basename(img2_path).split('.')[0]+'.png'
+        img1 = Image.open(img1_path)
+        img2 = Image.open(img2_path)
+        img1 = img1.resize((224,224))
+        img2 = img2.resize((224,224))
+        print('img1_path', img1)
+        print('img2_path', img2)
+        assert img1.height == img2.height
+
+        new_img = Image.new('RGB', (img1.width + img2.width + 10, img1.height))
+
+        # print('new_img', new_img)
+        new_img.paste(img1, (0, 0))
+        new_img.paste(img2, (img1.width + 10, 0))
+
+        if similar is not None:
+            similar = str(similar)+'_'+str(label)
+            draw = ImageDraw.Draw(new_img)
+            draw.text(position, str(similar), color, font_size=36)
+        os.makedirs(save_path, exist_ok=True)
+        img_save = os.path.join(save_path, img_name)
+        new_img.save(img_save)
+