blind_watermark.py 文件源码

def extract(self,ori_wmimage,wm,key=10):
        B = ori_wmimage
        if len(ori_wmimage.shape ) > 2 :
            (B,G,R) = cv2.split(cv2.cvtColor(ori_wmimage, cv2.COLOR_BGR2YUV))

        signature = BlindWatermark._gene_signature(wm,256,key).flatten()

        ext_sig = self.inner_extract(B,signature)
        return BlindWatermark.calc_sim(signature,ext_sig)