fix: gui button sizing & output saving

chompaa · Aug 9, 2023 · 21019eb · 21019eb
1 parent 110bbb6
commit 21019eb
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/main.py b/main.py
@@ -144,7 +144,6 @@ def __init__(self, image_width, image_height):
         select_image = QPushButton("Choose File")
         select_image.clicked.connect(self.select_image)
         select_image.setFixedWidth(self.image_width)
-        select_image.setFixedHeight(self.image_height)
 
         layout.addWidget(self.display_image, alignment=Qt.AlignmentFlag.AlignCenter)
         layout.addWidget(select_image, alignment=Qt.AlignmentFlag.AlignCenter)

diff --git a/test.py b/test.py
@@ -9,10 +9,6 @@
 from utils import convert_rgb_to_ycbcr, convert_ycbcr_to_rgb, calculate_psnr
 
 
-def get_image_name_from_path(path):
-    return path.split("/")[-1].split("\\")[-1]
-
-
 def sisr(weights_file, output_folder, image_file, scale, set_status):
     cudnn.benchmark = True
     device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
@@ -80,7 +76,7 @@ def sisr(weights_file, output_folder, image_file, scale, set_status):
 
     set_status("Saving SRCNN image...")
 
-    output.save(f"{output_folder}{image_file.replace('.', f'_srcnn_x{scale}.')}")
+    output.save(f"{output_folder}{image_name.replace('.', f'_srcnn_x{scale}.')}")
 
     set_status(f"Done! PSNR: {psnr:.2f}")