Update README

Author: Saint-lsy

LocalMatching.py

@@ -13,7 +13,7 @@
 
 
 def plot_matching_figure(image1, image2, xyA_list, xyB_list, save_path):
-  fig = plt.figure(figsize=(10,10))
+  fig = plt.figure(figsize=(11,5))
   ax1 = fig.add_subplot(121)
   ax2 = fig.add_subplot(122)
 
@@ -22,12 +22,15 @@ def plot_matching_figure(image1, image2, xyA_list, xyB_list, save_path):
 
   for xyA, xyB in zip(xyA_list, xyB_list):
     con = ConnectionPatch(xyA=xyB, xyB=xyA, coordsA="data", coordsB="data",
-                          axesA=ax2, axesB=ax1, color="red")
+                          axesA=ax2, axesB=ax1, color="green")
     ax2.add_artist(con)
 
   fig.tight_layout()
+  ax1.axis('off')
+  ax2.axis('off')
+  plt.subplots_adjust(wspace=0.05)
   fig.show()
-  fig.savefig(save_path)
+  fig.savefig(save_path, bbox_inches='tight', pad_inches=0, dpi=400)
 
 
 

README.md

@@ -1,4 +1,5 @@
-# Polyp-Gen
+# Polyp-Gen: Realistic and Diverse Polyp Image Generation for Endoscopic Dataset Expansion
+![introduction](assets/pipeline.jpg)
 
 ## Setup
 
@@ -25,7 +26,7 @@ You can train your own model using the script:
 bash scripts/train.sh
 ```
 ## Sampling
-
+![sampling](assets/showcase.jpg)
 ### Checkpoint
 You can download the chekpoints of our Polyp_Gen from [here](https://huggingface.co/Saint-lsy/Polyp-Gen-sd2-inpainting/tree/main).
 ### Sampling with Specified Mask
@@ -48,6 +49,8 @@ One Demo of LocalMatching
 python LocalMatching.py --ref_image demos/img_1513_neg.jpg --ref_mask  demos/mask_1513.jpg --query_image  demos/img_1592_neg.jpg --mask_proposal gen_mask.jpg
 ```
 
+<img src="assets/matching.jpg" alt="local_matching" width="450">
+
 The third step is using the generated Mask to sample.
 ## Acknowledgements
 The code is based on the following projects. Greatly thanks to these authors!