From 3f870167f336ec37a599cf7ee7e05964561cd828 Mon Sep 17 00:00:00 2001 From: Keycatowo Date: Sat, 20 Apr 2024 12:39:31 +0800 Subject: [PATCH] Fix logical error in image size check MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 在检查图片尺寸时，原本重复检查了两次长度，修正为长+宽 --- blind_watermark/att.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/blind_watermark/att.py b/blind_watermark/att.py index 040a9a5..731f12e 100644 --- a/blind_watermark/att.py +++ b/blind_watermark/att.py @@ -181,7 +181,7 @@ def anti_cut_att_old(input_filename, output_file_name, origin_shape): input_img = cv2.imread(input_filename) output_img = input_img.copy() output_img_shape = output_img.shape - if output_img_shape[0] > origin_shape[0] or output_img_shape[0] > origin_shape[0]: + if output_img_shape[0] > origin_shape[0] or output_img_shape[1] > origin_shape[1]: print('裁剪打击后的图片，不可能比原始图片大，检查一下') return @@ -204,7 +204,7 @@ def anti_cut_att(input_filename=None, input_img=None, output_file_name=None, ori input_img = cv2.imread(input_filename) output_img = input_img.copy() output_img_shape = output_img.shape - if output_img_shape[0] > origin_shape[0] or output_img_shape[0] > origin_shape[0]: + if output_img_shape[0] > origin_shape[0] or output_img_shape[1] > origin_shape[1]: print('裁剪打击后的图片，不可能比原始图片大，检查一下') return