截图后解码失败 about blind_watermark HOT 2 CLOSED

代码：

import numpy as np
from blind_watermark import WaterMark
from blind_watermark import att
from blind_watermark.recover import estimate_crop_parameters, recover_crop
import cv2
import os
import time

start_time=time.time()

bwm = WaterMark(password_img=1, password_wm=1)

bwm.read_img('examples/pic-test/NI7A7780_1.JPG')
wm = 'hello world.My name is xiaoming,my phone is 1237868765456'
bwm.read_wm(wm, mode='str')
bwm.embed('examples/output/embedded.png')

len_wm = len(bwm.wm_bit)
print('Put down the length of wm_bit {len_wm}'.format(len_wm=len_wm))

ori_img_shape = cv2.imread('examples/pic-test/NI7A7780_1.JPG').shape[:2]

bwm1 = WaterMark(password_img=1, password_wm=1)
wm_extract = bwm1.extract('examples/output/embedded.png', wm_shape=len_wm, mode='str')
print("不攻击的提取结果：", wm_extract)

try:
assert wm == wm_extract, '不攻击 提取水印和原水印不一致'
except AssertionError:
print('不攻击 提取水印和原水印不一致')

(x1, y1, x2, y2), image_o_shape, score, scale_infer = estimate_crop_parameters(original_file='examples/output/embedded.png',
template_file='examples/output/cut.png',
scale=(0.5, 2), search_num=200)

print(f'Crop attack's estimate parameters: x1={x1},y1={y1},x2={x2},y2={y2}. score={score}')

recover_crop(template_file='examples/output/cut.png', output_file_name='examples/output/jieping2_huanyuan.png',
loc=(x1, y1, x2, y2), image_o_shape=image_o_shape)

bwm1 = WaterMark(password_wm=1, password_img=1)
wm_extract = bwm1.extract('examples/output/jieping2_huanyuan.png', wm_shape=len_wm, mode='str')
print("截屏攻击，不知道攻击参数。提取结果：", wm_extract)
try:
assert wm == wm_extract, '截屏攻击，不知道攻击参数 提取水印和原水印不一致'
except AssertionError:
print('截屏攻击，不知道攻击参数 提取水印和原水印不一致')

from blind_watermark.

#81 (comment)

from blind_watermark.