iOS圖片相似度比較
阿新 • • 發佈：2017-08-23
整數 return per 計算 spa 獲取 wid last csb
1. 縮小尺寸：將圖像縮小到8*8的尺寸，總共64個像素。這一步的作用是去除圖像的細節，只保留結構/明暗等基本信息，摒棄不同尺寸/比例帶來的圖像差異；
註：實際操作時，采取了兩種尺寸作對比（10*10，100*100）尺寸再大的話就會性能影響就會較大了，我實現了兩種，目的是為了展示怎麽設定不同尺寸。
2. 簡化色彩：將縮小後的圖像，轉為64級灰度，即所有像素點總共只有64種顏色；
註：關於多少灰度級的問題，我並沒有太在意，采取了一個合適的RGB to GRAY 算法就好，個人理解
3. 計算平均值：計算所有64個像素的灰度平均值；
4. 比較像素的灰度：將每個像素的灰度，與平均值進行比較，大於或等於平均值記為1，小於平均值記為0；
5. 計算哈希值：將上一步的比較結果，組合在一起，就構成了一個64位的整數，這就是這張圖像的指紋。組合的次序並不重要，只要保證所有圖像都采用同樣次序就行了；
6. 得到指紋以後，就可以對比不同的圖像
GetSimilarity.h
// // GetSimilarity.h // imgsimlartest // // Created by test on 16/3/3. // Copyright ? 2016年 com.facishare.CoreTest. All rights reserved. // #import <Foundation/Foundation.h> #import <UIKit/UIKit.h> typedef double Similarity; @interface GetSimilarity : NSObject - (void)setImgWithImgA:(UIImage*)imgA ImgB:(UIImage*)imgB;//設置需要對比的圖片 - (void)setImgAWidthImg:(UIImage*)img; - (void)setImgBWidthImg:(UIImage*)img; - (Similarity)getSimilarityValue; //獲取相似度 + (Similarity)getSimilarityValueWithImgA:(UIImage*)imga ImgB:(UIImage*)imgb;//類方法 @end
GetSimilarity.m
//
// GetSimilarity.m
// imgsimlartest
//
// Created by test on 16/3/3.
// Copyright ? 2016年 com.facishare.CoreTest. All rights reserved.
//
#import "GetSimilarity.h"
#define ImgSizeA 10
#define ImgSizeB 100
typedef enum workday
{
SizeA,
SizeB,
}GetSimilarityType;
@interface GetSimilarity()
@property (nonatomic,assign) Similarity similarity;
@property (nonatomic,strong) UIImage *imga;
@property (nonatomic,strong) UIImage *imgb;
@end
@implementation GetSimilarity
- (instancetype)init
{
self = [super init];
if (self) {
self.imga = [[UIImage alloc]init];
self.imgb = [[UIImage alloc]init];
}
return self;
}
- (void)setImgWithImgA:(UIImage*)imgA ImgB:(UIImage*)imgB
{
_imga = imgA;
_imgb = imgB;
}
- (void)setImgAWidthImg:(UIImage*)img
{
self.imga = img;
}
- (void)setImgBWidthImg:(UIImage*)img
{
self.imgb = img;
}
- (Similarity)getSimilarityValue
{
self.similarity = MAX([self getSimilarityValueWithType:SizeA], [self getSimilarityValueWithType:SizeB]);
return self.similarity;
}
+ (Similarity)getSimilarityValueWithImgA:(UIImage *)imga ImgB:(UIImage *)imgb
{
GetSimilarity * getSimilarity = [[GetSimilarity alloc]init];
[getSimilarity setImgWithImgA:imga ImgB:imgb];
return [getSimilarity getSimilarityValue];
}
- (Similarity)getSimilarityValueWithType:(GetSimilarityType)type;//
{
int cursize = (type == SizeA ? ImgSizeA : ImgSizeB);
int ArrSize = cursize * cursize + 1,a[ArrSize],b[ArrSize],i,j,grey,sum = 0;
CGSize size = {cursize,cursize};
UIImage * imga = [self reSizeImage:self.imga toSize:size];
UIImage * imgb = [self reSizeImage:self.imgb toSize:size];//縮小圖片尺寸
a[ArrSize] = 0;
b[ArrSize] = 0;
CGPoint point;
for (i = 0 ; i < cursize; i++) {//計算a的灰度
for (j = 0; j < cursize; j++) {
point.x = i;
point.y = j;
grey = ToGrey([self UIcolorToRGB:[self colorAtPixel:point img:imga]]);
a[cursize * i + j] = grey;
a[ArrSize] += grey;
}
}
a[ArrSize] /= (ArrSize - 1);//灰度平均值
for (i = 0 ; i < cursize; i++) {//計算b的灰度
for (j = 0; j < cursize; j++) {
point.x = i;
point.y = j;
grey = ToGrey([self UIcolorToRGB:[self colorAtPixel:point img:imgb]]);
b[cursize * i + j] = grey;
b[ArrSize] += grey;
}
}
b[ArrSize] /= (ArrSize - 1);//灰度平均值
for (i = 0 ; i < ArrSize ; i++)//灰度分布計算
{
a[i] = (a[i] < a[ArrSize] ? 0 : 1);
b[i] = (b[i] < b[ArrSize] ? 0 : 1);
}
ArrSize -= 1;
for (i = 0 ; i < ArrSize ; i++)
{
sum += (a[i] == b[i] ? 1 : 0);
}
return sum * 1.0 / ArrSize;
}
- (UIImage *)reSizeImage:(UIImage *)image toSize:(CGSize)reSize//重新設定圖片尺寸
{
UIGraphicsBeginImageContext(CGSizeMake(reSize.width, reSize.height));
[image drawInRect:CGRectMake(0, 0, reSize.width, reSize.height)];
UIImage *reSizeImage = UIGraphicsGetImageFromCurrentImageContext();
UIGraphicsEndImageContext();
return reSizeImage;
}
unsigned int ToGrey(unsigned int rgb)//RGB計算灰度
{
unsigned int blue = (rgb & 0x000000FF) >> 0;
unsigned int green = (rgb & 0x0000FF00) >> 8;
unsigned int red = (rgb & 0x00FF0000) >> 16;
return ( red*38 + green * 75 + blue * 15 )>>7;
}
- (unsigned int)UIcolorToRGB:(UIColor*)color//UIColor轉16進制RGB
{
unsigned int RGB,R,G,B;
RGB = R = G = B = 0x00000000;
CGFloat r,g,b,a;
[color getRed:&r green:&g blue:&b alpha:&a];
R = r * 256 ;
G = g * 256 ;
B = b * 256 ;
RGB = (R << 16) | (G << 8) | B ;
return RGB;
}
- (UIColor *)colorAtPixel:(CGPoint)point img:(UIImage*)img{//獲取指定point位置的RGB
// Cancel if point is outside image coordinates
if (!CGRectContainsPoint(CGRectMake(0.0f, 0.0f, img.size.width, img.size.height), point)) { return nil; }
NSInteger pointX = trunc(point.x);
NSInteger pointY = trunc(point.y);
CGImageRef cgImage = img.CGImage;
NSUInteger width = img.size.width;
NSUInteger height = img.size.height;
int bytesPerPixel = 4;
int bytesPerRow = bytesPerPixel * 1;
NSUInteger bitsPerComponent = 8;
unsigned char pixelData[4] = { 0, 0, 0, 0 };
CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
CGContextRef context = CGBitmapContextCreate(pixelData, 1, 1, bitsPerComponent, bytesPerRow, colorSpace, kCGImageAlphaPremultipliedLast | kCGBitmapByteOrder32Big);
CGColorSpaceRelease(colorSpace);
CGContextSetBlendMode(context, kCGBlendModeCopy);
// Draw the pixel we are interested in onto the bitmap context
CGContextTranslateCTM(context, -pointX, pointY-(CGFloat)height);
CGContextDrawImage(context, CGRectMake(0.0f, 0.0f, (CGFloat)width, (CGFloat)height), cgImage);
CGContextRelease(context);
// Convert color values [0..255] to floats [0.0..1.0]
CGFloat red = (CGFloat)pixelData[0] / 255.0f;
CGFloat green = (CGFloat)pixelData[1] / 255.0f;
CGFloat blue = (CGFloat)pixelData[2] / 255.0f;
CGFloat alpha = (CGFloat)pixelData[3] / 255.0f;
return [UIColor colorWithRed:red green:green blue:blue alpha:alpha];
}
@end
iOS圖片相似度比較