forked from BradLarson/GPUImage
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Added a color-based local binary pattern filter.
- Loading branch information
1 parent
25c8146
commit b421e8a
Showing
9 changed files
with
201 additions
and
3 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,5 @@ | ||
| #import "GPUImage3x3TextureSamplingFilter.h" | ||
|
|
||
| @interface GPUImageColorLocalBinaryPatternFilter : GPUImage3x3TextureSamplingFilter | ||
|
|
||
| @end |
159 changes: 159 additions & 0 deletions
159
framework/Source/GPUImageColorLocalBinaryPatternFilter.m
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,159 @@ | ||
| #import "GPUImageColorLocalBinaryPatternFilter.h" | ||
|
|
||
| // This is based on "Accelerating image recognition on mobile devices using GPGPU" by Miguel Bordallo Lopez, Henri Nykanen, Jari Hannuksela, Olli Silven and Markku Vehvilainen | ||
| // http://www.ee.oulu.fi/~jhannuks/publications/SPIE2011a.pdf | ||
|
|
||
| // Right pixel is the most significant bit, traveling clockwise to get to the upper right, which is the least significant | ||
| // If the external pixel is greater than or equal to the center, set to 1, otherwise 0 | ||
| // | ||
| // 2 1 0 | ||
| // 3 7 | ||
| // 4 5 6 | ||
|
|
||
| // 01101101 | ||
| // 76543210 | ||
|
|
||
| @implementation GPUImageColorLocalBinaryPatternFilter | ||
|
|
||
| #if TARGET_IPHONE_SIMULATOR || TARGET_OS_IPHONE | ||
| NSString *const kGPUImageColorLocalBinaryPatternFragmentShaderString = SHADER_STRING | ||
| ( | ||
| precision highp float; | ||
|
|
||
| varying vec2 textureCoordinate; | ||
| varying vec2 leftTextureCoordinate; | ||
| varying vec2 rightTextureCoordinate; | ||
|
|
||
| varying vec2 topTextureCoordinate; | ||
| varying vec2 topLeftTextureCoordinate; | ||
| varying vec2 topRightTextureCoordinate; | ||
|
|
||
| varying vec2 bottomTextureCoordinate; | ||
| varying vec2 bottomLeftTextureCoordinate; | ||
| varying vec2 bottomRightTextureCoordinate; | ||
|
|
||
| uniform sampler2D inputImageTexture; | ||
|
|
||
| void main() | ||
| { | ||
| lowp vec3 centerColor = texture2D(inputImageTexture, textureCoordinate).rgb; | ||
| lowp vec3 bottomLeftColor = texture2D(inputImageTexture, bottomLeftTextureCoordinate).rgb; | ||
| lowp vec3 topRightColor = texture2D(inputImageTexture, topRightTextureCoordinate).rgb; | ||
| lowp vec3 topLeftColor = texture2D(inputImageTexture, topLeftTextureCoordinate).rgb; | ||
| lowp vec3 bottomRightColor = texture2D(inputImageTexture, bottomRightTextureCoordinate).rgb; | ||
| lowp vec3 leftColor = texture2D(inputImageTexture, leftTextureCoordinate).rgb; | ||
| lowp vec3 rightColor = texture2D(inputImageTexture, rightTextureCoordinate).rgb; | ||
| lowp vec3 bottomColor = texture2D(inputImageTexture, bottomTextureCoordinate).rgb; | ||
| lowp vec3 topColor = texture2D(inputImageTexture, topTextureCoordinate).rgb; | ||
|
|
||
| lowp float redByteTally = 1.0 / 255.0 * step(centerColor.r, topRightColor.r); | ||
| redByteTally += 2.0 / 255.0 * step(centerColor.r, topColor.r); | ||
| redByteTally += 4.0 / 255.0 * step(centerColor.r, topLeftColor.r); | ||
| redByteTally += 8.0 / 255.0 * step(centerColor.r, leftColor.r); | ||
| redByteTally += 16.0 / 255.0 * step(centerColor.r, bottomLeftColor.r); | ||
| redByteTally += 32.0 / 255.0 * step(centerColor.r, bottomColor.r); | ||
| redByteTally += 64.0 / 255.0 * step(centerColor.r, bottomRightColor.r); | ||
| redByteTally += 128.0 / 255.0 * step(centerColor.r, rightColor.r); | ||
|
|
||
| lowp float blueByteTally = 1.0 / 255.0 * step(centerColor.b, topRightColor.b); | ||
| blueByteTally += 2.0 / 255.0 * step(centerColor.b, topColor.b); | ||
| blueByteTally += 4.0 / 255.0 * step(centerColor.b, topLeftColor.b); | ||
| blueByteTally += 8.0 / 255.0 * step(centerColor.b, leftColor.b); | ||
| blueByteTally += 16.0 / 255.0 * step(centerColor.b, bottomLeftColor.b); | ||
| blueByteTally += 32.0 / 255.0 * step(centerColor.b, bottomColor.b); | ||
| blueByteTally += 64.0 / 255.0 * step(centerColor.b, bottomRightColor.b); | ||
| blueByteTally += 128.0 / 255.0 * step(centerColor.b, rightColor.b); | ||
|
|
||
| lowp float greenByteTally = 1.0 / 255.0 * step(centerColor.g, topRightColor.g); | ||
| greenByteTally += 2.0 / 255.0 * step(centerColor.g, topColor.g); | ||
| greenByteTally += 4.0 / 255.0 * step(centerColor.g, topLeftColor.g); | ||
| greenByteTally += 8.0 / 255.0 * step(centerColor.g, leftColor.g); | ||
| greenByteTally += 16.0 / 255.0 * step(centerColor.g, bottomLeftColor.g); | ||
| greenByteTally += 32.0 / 255.0 * step(centerColor.g, bottomColor.g); | ||
| greenByteTally += 64.0 / 255.0 * step(centerColor.g, bottomRightColor.g); | ||
| greenByteTally += 128.0 / 255.0 * step(centerColor.g, rightColor.g); | ||
|
|
||
| // TODO: Replace the above with a dot product and two vec4s | ||
| // TODO: Apply step to a matrix, rather than individually | ||
|
|
||
| gl_FragColor = vec4(redByteTally, blueByteTally, greenByteTally, 1.0); | ||
| } | ||
| ); | ||
| #else | ||
| NSString *const kGPUImageColorLocalBinaryPatternFragmentShaderString = SHADER_STRING | ||
| ( | ||
| varying vec2 textureCoordinate; | ||
| varying vec2 leftTextureCoordinate; | ||
| varying vec2 rightTextureCoordinate; | ||
|
|
||
| varying vec2 topTextureCoordinate; | ||
| varying vec2 topLeftTextureCoordinate; | ||
| varying vec2 topRightTextureCoordinate; | ||
|
|
||
| varying vec2 bottomTextureCoordinate; | ||
| varying vec2 bottomLeftTextureCoordinate; | ||
| varying vec2 bottomRightTextureCoordinate; | ||
|
|
||
| uniform sampler2D inputImageTexture; | ||
|
|
||
| void main() | ||
| { | ||
| vec3 centerColor = texture2D(inputImageTexture, textureCoordinate).rgb; | ||
| vec3 bottomLeftColor = texture2D(inputImageTexture, bottomLeftTextureCoordinate).rgb; | ||
| vec3 topRightColor = texture2D(inputImageTexture, topRightTextureCoordinate).rgb; | ||
| vec3 topLeftColor = texture2D(inputImageTexture, topLeftTextureCoordinate).rgb; | ||
| vec3 bottomRightColor = texture2D(inputImageTexture, bottomRightTextureCoordinate).rgb; | ||
| vec3 leftColor = texture2D(inputImageTexture, leftTextureCoordinate).rgb; | ||
| vec3 rightColor = texture2D(inputImageTexture, rightTextureCoordinate).rgb; | ||
| vec3 bottomColor = texture2D(inputImageTexture, bottomTextureCoordinate).rgb; | ||
| vec3 topColor = texture2D(inputImageTexture, topTextureCoordinate).rgb; | ||
|
|
||
| float redByteTally = 1.0 / 255.0 * step(centerColor.r, topRightColor.r); | ||
| redByteTally += 2.0 / 255.0 * step(centerColor.r, topColor.r); | ||
| redByteTally += 4.0 / 255.0 * step(centerColor.r, topLeftColor.r); | ||
| redByteTally += 8.0 / 255.0 * step(centerColor.r, leftColor.r); | ||
| redByteTally += 16.0 / 255.0 * step(centerColor.r, bottomLeftColor.r); | ||
| redByteTally += 32.0 / 255.0 * step(centerColor.r, bottomColor.r); | ||
| redByteTally += 64.0 / 255.0 * step(centerColor.r, bottomRightColor.r); | ||
| redByteTally += 128.0 / 255.0 * step(centerColor.r, rightColor.r); | ||
|
|
||
| float blueByteTally = 1.0 / 255.0 * step(centerColor.b, topRightColor.b); | ||
| blueByteTally += 2.0 / 255.0 * step(centerColor.b, topColor.b); | ||
| blueByteTally += 4.0 / 255.0 * step(centerColor.b, topLeftColor.b); | ||
| blueByteTally += 8.0 / 255.0 * step(centerColor.b, leftColor.b); | ||
| blueByteTally += 16.0 / 255.0 * step(centerColor.b, bottomLeftColor.b); | ||
| blueByteTally += 32.0 / 255.0 * step(centerColor.b, bottomColor.b); | ||
| blueByteTally += 64.0 / 255.0 * step(centerColor.b, bottomRightColor.b); | ||
| blueByteTally += 128.0 / 255.0 * step(centerColor.b, rightColor.b); | ||
|
|
||
| float greenByteTally = 1.0 / 255.0 * step(centerColor.g, topRightColor.g); | ||
| greenByteTally += 2.0 / 255.0 * step(centerColor.g, topColor.g); | ||
| greenByteTally += 4.0 / 255.0 * step(centerColor.g, topLeftColor.g); | ||
| greenByteTally += 8.0 / 255.0 * step(centerColor.g, leftColor.g); | ||
| greenByteTally += 16.0 / 255.0 * step(centerColor.g, bottomLeftColor.g); | ||
| greenByteTally += 32.0 / 255.0 * step(centerColor.g, bottomColor.g); | ||
| greenByteTally += 64.0 / 255.0 * step(centerColor.g, bottomRightColor.g); | ||
| greenByteTally += 128.0 / 255.0 * step(centerColor.g, rightColor.g); | ||
|
|
||
| // TODO: Replace the above with a dot product and two vec4s | ||
| // TODO: Apply step to a matrix, rather than individually | ||
|
|
||
| gl_FragColor = vec4(redByteTally, blueByteTally, greenByteTally, 1.0); | ||
| } | ||
| ); | ||
| #endif | ||
|
|
||
| #pragma mark - | ||
| #pragma mark Initialization and teardown | ||
|
|
||
| - (id)init; | ||
| { | ||
| if (!(self = [super initWithFragmentShaderFromString:kGPUImageColorLocalBinaryPatternFragmentShaderString])) | ||
| { | ||
| return nil; | ||
| } | ||
|
|
||
| return self; | ||
| } | ||
|
|
||
| @end |
Oops, something went wrong.