前阵子在阮一峰的博客上看到了这篇《相似图片搜索原理》博客,就有一种冲动要将这些原理实现出来了。
Google "相似图片搜索":你可以用一张图片,搜索互联网上所有与它相似的图片。
打开Google图片搜索页面:
点击
使用上传一张angelababy原图:
点击搜索后,Google将会找出与之相似的图片,图片相似度越高就越排在前面。如:
这种技术的原理是什么?计算机怎么知道两张图片相似呢?
根据Neal Krawetz博士的解释,实现相似图片搜素的关键技术叫做"感知哈希算法"(Perceptualhash algorithm),它的作用是对每张图片生成一个"指纹"(fingerprint)字符串,然后比较不同图片的指纹。结果越接近,就说明图片越相似。
以下是一个最简单的Java实现:
预处理:读取图片
1. FileinputFile=newFile(filename);
2. //读取图片文件
第一步,缩小尺寸。
将图片缩小到8x8的尺寸,总共64个像素。这一步的作用是去除图片的细节,只保留结构、明暗等基本信息,摒弃不同尺寸、比例带来的图片差异。
int width= 8; intheight = 8; // targetW,targetH分别表示目标长和宽 int type= sourceImage.getType();// 图片类型 BufferedImagethumbImage = null; double sx= (double) width / sourceImage.getWidth(); double sy= (double) height / sourceImage.getHeight();
1. //将图片宽度和高度都设置成一样,以长度短的为准
2. if(b){
3. if(sx>sy){
4. sx=sy;
5. int)(sx*sourceImage.getWidth());
6. else{
7. sy=sx;
8. int)(sy*sourceImage.getHeight());
9. }
10. }
11. //自定义图片
12. if(type==BufferedImage.TYPE_CUSTOM){//handmade
13. ColorModelcm=sourceImage.getColorModel();
14. WritableRasterraster=cm.createCompatibleWritableRaster(width,height);
15. booleanalphaPremultiplied=cm.isAlphaPremultiplied();
16. newBufferedImage(cm,raster,alphaPremultiplied,null);
17. else{
18. //已知图片,如jpg,png,gif
19. newBufferedImage(width,height,type);
20. }
21. //调用画图类画缩小尺寸后的图
22. Graphics2Dg=target.createGraphics();
23. //smootherthanexlax:
24. g.setRenderingHint(RenderingHints.KEY_RENDERING,RenderingHints.VALUE_RENDER_QUALITY);
25. g.drawRenderedImage(sourceImage,AffineTransform.getScaleInstance(sx,sy));
26. g.dispose();
第二步,简化色彩。
将缩小后的图片,转为64级灰度。也就是说,所有像素点总共只有64种颜色。
1. int[]pixels=newint[width*height];
2. inti=0;i<width;i++){
3. j=0;j<height;j++){
4. pixels[i*height+j]=rgbToGray(thumbImage.getRGB(i,j));
5. }
6. }
7. /**
8. *灰度值计算
9. *@parampixels彩色RGB值(Red-Green-Blue红绿蓝)
10. *@returnint灰度值
11. */
12. publicstaticintrgbToGray(intpixels){
13. _alpha=(pixels>>24)&0xFF;
14. _red=(pixels>>16)&0xFF;
15. _green=(pixels>>8)&0xFF;
16. _blue=(pixels)&0xFF;
17. return(int)(0.3*_red+0.59*_green+0.11*_blue);
18. }
第三步,计算平均值。
计算所有64个像素的灰度平均值。
1. intavgPixel=0;
2. intm=0;
3. for(inti=0;i<pixels.length;++i){
4. m+=pixels[i];
5. }
6. m=m/pixels.length;
7. avgPixel=m;
第四步,比较像素的灰度。
将每个像素的灰度,与平均值进行比较。大于或等于平均值,记为1;小于平均值,记为0。
1. int[]comps=newint[width*height];
2. for(inti=0;i<comps.length;i++){
3. if(pixels[i]>=avgPixel){
4. 1;
5. else{
6. 0;
7. }
8. }
第五步,计算哈希值。
将上一步的比较结果,组合在一起,就构成了一个64位的整数,这就是这张图片的指纹。组合的次序并不重要,只要保证所有图片都采用同样次序就行了。
=
= 8f373714acfcf4d0
1. StringBufferhashCode=newStringBuffer();
2. inti=0;i<comps.length;i+=4){
3. intresult=comps[i]*(int)Math.pow(2,3)+comps[i+1]*(int)Math.pow(2,2)+comps[i+2]*(int)Math.pow(2,1)+comps[i+2];
4. hashCode.append(binaryToHex(result));//二进制转为16进制
5. }
6. StringsourceHashCode=hashCode.toString();
得到指纹以后,就可以对比不同的图片,看看64位中有多少位是不一样的。在理论上,这等同于计算"汉明距离"(Hammingdistance)。如果不相同的数据位不超过5,就说明两张图片很相似;如果大于10,就说明这是两张不同的图片。
1. intdifference=0;
2. intlen=sourceHashCode.length();
3.
4. for(inti=0;i<len;i++){
5. if(sourceHashCode.charAt(i)!=hashCode.charAt(i)){
6. difference++;
7. }
8. }
你可以将几张图片放在一起,也计算出他们的汉明距离对比,就可以看看两张图片是否相似。
这种算法的优点是简单快速,不受图片大小缩放的影响,缺点是图片的内容不能变更。如果在图片上加几个文字,它就认不出来了。所以,它的最佳用途是根据缩略图,找出原图。
实际应用中,往往采用更强大的pHash算法和SIFT算法,它们能够识别图片的变形。只要变形程度不超过25%,它们就能匹配原图。这些算法虽然更复杂,但是原理与上面的简便算法是一样的,就是先将图片转化成Hash字符串,然后再进行比较。
以上内容大部分直接从阮一峰的网站上复制过来,想看原著的童鞋可以去在最上面的链接点击进去看。