感知哈希（pHash）和梯度哈希（dHash）的Python+OpenCV+SciPy实现

最新推荐文章于 2024-08-07 08:26:05 发布

ZhuTH

最新推荐文章于 2024-08-07 08:26:05 发布

阅读量2k

点赞数

文章标签： opencv python

本文链接：https://blog.csdn.net/ZhuTH/article/details/46480395

版权

</pre><pre name="code" class="python">class ImageUtils:
   
    @staticmethod
    def cvload(buf):
        npa = np.asarray(bytearray(buf), dtype=np.uint8)
        im = cv2.imdecode(npa, 1)
        return im
        
    @staticmethod
    def cvencode(im):
        ret, buf = cv2.imencode('.jpg', im)
        buf = np.array(buf).tostring()
        return buf
        
    @staticmethod
    def cvgrey(im):
        im = cv2.cvtColor(im, cv2.COLOR_BGR2GRAY)
        return im

    @staticmethod
    def phash(im):
        if im is None:
            return 0
        if len(im.shape) == 3 and im.shape[-1] == 3:
            im = ImageUtils.cvgrey(im)
        im = im.astype(float)
        im = ImageUtils.cvresize(im, (32, 32))
        from scipy.fftpack import dct
        dim = dct(im)
        mean = np.sum(dim) / (32. * 32.)
        val = 0