MVP

snucvpip · Dec 9, 2021 · ba8b647 · ba8b647
1 parent 407c2d2
commit ba8b647
Show file tree

Hide file tree

Showing 48 changed files with 153 additions and 140 deletions.
diff --git a/images_to_crop.png b/images_to_crop.png
diff --git a/impl/images_to_crop.png b/impl/images_to_crop.png
diff --git a/impl/main.ipynb b/impl/main.ipynb
diff --git a/impl/post_process/eval.py b/impl/post_process/eval.py
@@ -136,7 +136,7 @@ def CropImage(im, imReference, tolerance=100):
     ax[1].imshow(imReference)
     ax[1].scatter(vimReference[:,0], vimReference[:,1], marker='x', c='red', s=200, linewidth=3)
     plt.tight_layout()
-    plt.savefig('images_to_crop.png', dpi=300)    
+    plt.savefig('images_to_crop.png', dpi=300, bbox_inches="tight")    
     plt.close()
 
     H, _ = cv2.findHomography(vim, vimReference)
@@ -178,20 +178,21 @@ def main(source, target, clean, resultdir, tol=100):
     tar = cv2.cvtColor(tar, cv2.COLOR_BGR2RGB)
 
     plt.close()
-    fig, ax = plt.subplots(ncols=3, figsize=(15, 10))
-    print(f'    Source\tMSE: {mse_src:.2f}\tPSNR: {psnr_src:.2f}\tSSIM: {ssim_src:.2f}')
-    ax[0].imshow(src), ax[0].set_title('Source')
-    ax[1].imshow(tar), ax[1].set_title('Target')
-    ax[2].imshow(diff_src, cmap='gray', vmin=0, vmax=255), ax[2].set_title('Difference')
-    plt.tight_layout()
-    plt.savefig(os.path.join(resultdir, 'source_result.pdf'), bbox_inches="tight")
 
-    plt.close()
-    fig, ax = plt.subplots(ncols=3, figsize=(15, 10))
-    print(f'    Clear\tMSE: {mse_cle:.2f}\tPSNR: {psnr_cle:.2f}\tSSIM: {ssim_cle:.2f}')
-    ax[0].imshow(cle), ax[0].set_title('Clean')
-    ax[1].imshow(tar), ax[1].set_title('Target')
-    ax[2].imshow(diff_cle, cmap='gray', vmin=0, vmax=255), ax[2].set_title('Difference')
+    fig, ax = plt.subplots(ncols=4, figsize=(15, 10))
+    ax[0].imshow(src)
+    ax[1].imshow(cle)
+    ax[2].imshow(tar)
+    ax[3].imshow(diff_cle, cmap='gray', vmin=0, vmax=255)
+
+    ax[0].set_title('Source')
+    ax[1].set_title('Clean')
+    ax[2].set_title('Target')
+    ax[3].set_title('Difference')
+
     plt.tight_layout()
-    plt.savefig(os.path.join(resultdir, 'clean_result.pdf'), bbox_inches="tight")
+    plt.savefig(os.path.join(resultdir, 'result.pdf'), bbox_inches="tight")
+
+    print(f'    Source\tMSE: {mse_src:.2f}\tPSNR: {psnr_src:.2f}\tSSIM: {ssim_src:.2f}')
+    print(f'    Clean\tMSE: {mse_cle:.2f}\tPSNR: {psnr_cle:.2f}\tSSIM: {ssim_cle:.2f}')
 
diff --git a/output.png b/output.png
diff --git a/result/graph_desktop/clean_result.pdf b/result/graph_desktop/clean_result.pdf
diff --git a/result/graph_desktop/result.pdf b/result/graph_desktop/result.pdf
diff --git a/result/graph_desktop/source_result.pdf b/result/graph_desktop/source_result.pdf
diff --git a/result/graph_laptop/clean_result.pdf b/result/graph_laptop/clean_result.pdf
diff --git a/result/graph_laptop/result.pdf b/result/graph_laptop/result.pdf
diff --git a/result/graph_laptop/source_result.pdf b/result/graph_laptop/source_result.pdf
diff --git a/result/intro_desktop/clean_result.pdf b/result/intro_desktop/clean_result.pdf
diff --git a/result/intro_desktop/result.pdf b/result/intro_desktop/result.pdf
diff --git a/result/intro_desktop/source_result.pdf b/result/intro_desktop/source_result.pdf
diff --git a/result/intro_laptop/clean_result.pdf b/result/intro_laptop/clean_result.pdf
diff --git a/result/intro_laptop/result.pdf b/result/intro_laptop/result.pdf
diff --git a/result/intro_laptop/source_result.pdf b/result/intro_laptop/source_result.pdf
diff --git a/result/lecture_desktop/clean_result.pdf b/result/lecture_desktop/clean_result.pdf
diff --git a/result/lecture_desktop/result.pdf b/result/lecture_desktop/result.pdf
diff --git a/result/lecture_desktop/source_result.pdf b/result/lecture_desktop/source_result.pdf
diff --git a/result/lecture_laptop/clean_result.pdf b/result/lecture_laptop/clean_result.pdf
diff --git a/result/lecture_laptop/result.pdf b/result/lecture_laptop/result.pdf
diff --git a/result/lecture_laptop/source_result.pdf b/result/lecture_laptop/source_result.pdf
diff --git a/result/physics/clean_cropped.png b/result/physics/clean_cropped.png
diff --git a/result/physics/clean_result.pdf b/result/physics/clean_result.pdf
diff --git a/result/physics/result.pdf b/result/physics/result.pdf
diff --git a/result/physics/source_cropped.png b/result/physics/source_cropped.png
diff --git a/result/physics/source_result.pdf b/result/physics/source_result.pdf
diff --git a/result/sample/clean_result.pdf b/result/sample/clean_result.pdf
diff --git a/result/sample/result.pdf b/result/sample/result.pdf
diff --git a/result/sample/source_result.pdf b/result/sample/source_result.pdf
diff --git a/result/snipping/clean_cropped.png b/result/snipping/clean_cropped.png
diff --git a/result/snipping/clean_result.pdf b/result/snipping/clean_result.pdf
diff --git a/result/snipping/result.pdf b/result/snipping/result.pdf
diff --git a/result/snipping/source_cropped.png b/result/snipping/source_cropped.png
diff --git a/result/snipping/source_result.pdf b/result/snipping/source_result.pdf
diff --git a/result/table_desktop/clean_result.pdf b/result/table_desktop/clean_result.pdf
diff --git a/result/table_desktop/result.pdf b/result/table_desktop/result.pdf
diff --git a/result/table_desktop/source_result.pdf b/result/table_desktop/source_result.pdf
diff --git a/result/table_laptop/clean_result.pdf b/result/table_laptop/clean_result.pdf
diff --git a/result/table_laptop/result.pdf b/result/table_laptop/result.pdf
diff --git a/result/table_laptop/source_result.pdf b/result/table_laptop/source_result.pdf
diff --git a/result/text_desktop/clean_result.pdf b/result/text_desktop/clean_result.pdf
diff --git a/result/text_desktop/result.pdf b/result/text_desktop/result.pdf
diff --git a/result/text_desktop/source_result.pdf b/result/text_desktop/source_result.pdf
diff --git a/result/text_laptop/clean_result.pdf b/result/text_laptop/clean_result.pdf
diff --git a/result/text_laptop/result.pdf b/result/text_laptop/result.pdf
diff --git a/result/text_laptop/source_result.pdf b/result/text_laptop/source_result.pdf