[TESTING] Removing spaces and padding in results images

hbenbel · Jan 19, 2021 · eba8738 · eba8738
1 parent 46bd828
commit eba8738
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/colorize.py b/colorize.py
@@ -71,7 +71,7 @@ def main(config):
     ax2.imshow(post_processed)
     ax2.axis('off')
 
-    fig.savefig(save_path)
+    fig.savefig(save_path, bbox_inches='tight', pad_inches=0)
 
 
 if __name__ == "__main__":

diff --git a/src/tester/tester.py b/src/tester/tester.py
@@ -49,7 +49,11 @@ def _save_result(self, original_image, predicted_image):
         ax2.imshow(predicted_image)
         ax2.axis('off')
 
-        fig.savefig(os.path.join(test_save_path, str(self.cpt) + '.png'))
+        fig.savefig(
+            os.path.join(test_save_path, str(self.cpt) + '.png'),
+            bbox_inches='tight',
+            pad_inches=0
+        )
 
         self.cpt += 1