watermark.py 文件源码

def channel_embedding(origin_image_chan, watermark_img_chan):
    coeffs1_1, coeffs1_2, coeffs1_3, coeffs2_3 = dwt2(origin_image_chan, watermark_img_chan)
    embedding_image = cv2.add(cv2.multiply(ORIGIN_RATE, coeffs1_3[0]), cv2.multiply(WATERMARK_RATE, coeffs2_3[0]))
    embedding_image = idwt2(embedding_image, coeffs1_1[1], coeffs1_2[1], coeffs1_3[1])
    np.clip(embedding_image, 0, 255, out=embedding_image)
    embedding_image = embedding_image.astype('uint8')
    return embedding_image