diff --git a/main.py b/main.py index 65534fd..0926d77 100644 --- 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 index 49a5b4f..9763ab1 100644 --- 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}")