PHP实现的简易版图片相似度比较

PHP实现的简易版图片相似度比较

由于相似图片搜索的php实现的 API 不怎么符合我的用途，所以我重新定义 API 的架构，改写成比较简单的函数方式，虽然还是用对象的方式包装。

代码如下:

<?php
/**
* 图片相似度比较
*
* @version $Id: ImageHash.php 4429 2012-04-17 13:20:31Z jax $
* @author jax.hu
*
*
* //Sample_1
* $aHash = ImageHash::hashImageFile('wsz.11.jpg');
* $bHash = ImageHash::hashImageFile('wsz.12.jpg');
* var_dump(ImageHash::isHashSimilar($aHash, $bHash));
*
* //Sample_2
* var_dump(ImageHash::isImageFileSimilar('wsz.11.jpg', 'wsz.12.jpg'));
*
*/

class ImageHash {

/**取样倍率 1~10
* @access public
* @staticvar int
* */
public static $rate = 2;

/**相似度允许值 0~64
* @access public
* @staticvar int
* */
public static $similarity = 80;

/**图片类型对应的开启函数
* @access private
* @staticvar string
* */
private static $_createFunc = array(
IMAGETYPE_GIF =>'imageCreateFromGIF',
IMAGETYPE_JPEG =>'imageCreateFromJPEG',
IMAGETYPE_PNG =>'imageCreateFromPNG',
IMAGETYPE_BMP =>'imageCreateFromBMP',
IMAGETYPE_WBMP =>'imageCreateFromWBMP',
IMAGETYPE_XBM =>'imageCreateFromXBM',
);


/**从文件建立图片
* @param string $filePath 文件地址路径
* @return resource 当成功开启图片则传递图片 resource ID，失败则是 false
* */
public static function createImage($filePath){
if(!file_exists($filePath)){ return false; }

/*判断文件类型是否可以开启*/
$type = exif_imagetype($filePath);
if(!array_key_exists($type,self::$_createFunc)){ return false; }

$func = self::$_createFunc[$type];
if(!function_exists($func)){ return false; }

return $func($filePath);
}


/**hash 图片
* @param resource $src 图片 resource ID
* @return string 图片 hash 值，失败则是 false
* */
public static function hashImage($src){
if(!$src){ return false; }

/*缩小图片尺寸*/
$delta = 8 * self::$rate;
$img = imageCreateTrueColor($delta,$delta);
imageCopyResized($img,$src, 0,0,0,0, $delta,$delta,imagesX($src),imagesY($src));

/*计算图片灰阶值*/
$grayArray = array();
for ($y=0; $y<$delta; $y++){
for ($x=0; $x<$delta; $x++){
$rgb = imagecolorat($img,$x,$y);
$col = imagecolorsforindex($img, $rgb);
$gray = intval(($col['red']+$col['green']+$col['blue'])/3)& 0xFF;

$grayArray[] = $gray;
}
}
imagedestroy($img);

/*计算所有像素的灰阶平均值*/
$average = array_sum($grayArray)/count($grayArray);

/*计算 hash 值*/
$hashStr = '';
foreach ($grayArray as $gray){
$hashStr .= ($gray>=$average) ? '1' : '0';
}

return $hashStr;
}


/**hash 图片文件
* @param string $filePath 文件地址路径
* @return string 图片 hash 值，失败则是 false
* */
public static function hashImageFile($filePath){
$src = self::createImage($filePath);
$hashStr = self::hashImage($src);
imagedestroy($src);

return $hashStr;
}


/**比较两个 hash 值，是不是相似
* @param string $aHash A图片的 hash 值
* @param string $bHash B图片的 hash 值
* @return bool 当图片相似则传递 true，否则是 false
* */
public static function isHashSimilar($aHash, $bHash){
$aL = strlen($aHash); $bL = strlen($bHash);
if ($aL !== $bL){ return false; }

/*计算容许落差的数量*/
$allowGap = $aL*(100-self::$similarity)/100;

/*计算两个 hash 值的汉明距离*/
$distance = 0;
for($i=0; $i<$aL; $i++){
if ($aHash{$i} !== $bHash{$i}){ $distance++; }
}

return ($distance<=$allowGap) ? true : false;
}


/**比较两个图片文件，是不是相似
* @param string $aHash A图片的路径
* @param string $bHash B图片的路径
* @return bool 当图片相似则传递 true，否则是 false
* */
public static function isImageFileSimilar($aPath, $bPath){
$aHash = ImageHash::hashImageFile($aPath);
$bHash = ImageHash::hashImageFile($bPath);
return ImageHash::isHashSimilar($aHash, $bHash);
}

}