channels/private/gradientMex.cpp: improved boundary handling for grad…

…Hist() softBin%2==1
JasonWay · Aug 20, 2013 · 8f1a978 · 8f1a978
1 parent 9e42261
commit 8f1a978
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/channels/private/gradientMex.cpp b/channels/private/gradientMex.cpp
@@ -221,6 +221,13 @@ void gradHist( float *M, float *O, float *H, int h, int w,
     }
   }
   alFree(O0); alFree(O1); alFree(M0); alFree(M1);
+  // normalize boundary bins which only get 7/8 of weight of interior bins
+  if( softBin%2!=0 ) for( int o=0; o<nOrients; o++ ) {
+    x=0; for( y=0; y<hb; y++ ) H[o*nb+x*hb+y]*=8.f/7.f;
+    y=0; for( x=0; x<wb; x++ ) H[o*nb+x*hb+y]*=8.f/7.f;
+    x=wb-1; for( y=0; y<hb; y++ ) H[o*nb+x*hb+y]*=8.f/7.f;
+    y=hb-1; for( x=0; x<wb; x++ ) H[o*nb+x*hb+y]*=8.f/7.f;
+  }
 }
 
 /******************************************************************************/

diff --git a/channels/private/gradientMex.mexw64 b/channels/private/gradientMex.mexw64