Commit 14d34ec9 authored by Davis E. King's avatar Davis E. King

Merge pull request #89 from rayglover/dev

introduce a configurable threshold for rendering HOGs
parents 7f611ea7 98cd2a92
......@@ -1058,7 +1058,8 @@ namespace dlib
>
matrix<unsigned char> draw_fhog(
const dlib::array<array2d<T,mm1>,mm2>& hog,
const long cell_draw_size = 15
const long cell_draw_size = 15,
const float min_response_threshold = 0.0
)
{
// make sure requires clause is not broken
......@@ -1084,7 +1085,7 @@ namespace dlib
const float val = hog[d][r/cell_draw_size][c/cell_draw_size] +
hog[d+mbars.size()][r/cell_draw_size][c/cell_draw_size] +
hog[d+mbars.size()*2][r/cell_draw_size][c/cell_draw_size];
if (val > 0)
if (val > min_response_threshold)
{
set_subm(himg, r, c, cell_draw_size, cell_draw_size) += val*mbars[d%mbars.size()];
}
......@@ -1106,7 +1107,8 @@ namespace dlib
>
matrix<unsigned char> draw_fhog (
const std::vector<matrix<T> >& hog,
const long cell_draw_size = 15
const long cell_draw_size = 15,
const float min_response_threshold = 0.0
)
{
// make sure requires clause is not broken
......@@ -1131,7 +1133,7 @@ namespace dlib
}
}
}
return draw_fhog(temp,cell_draw_size);
return draw_fhog(temp,cell_draw_size, min_response_threshold);
}
// ----------------------------------------------------------------------------------------
......@@ -1142,7 +1144,8 @@ namespace dlib
>
matrix<unsigned char> draw_fhog(
const array2d<matrix<T,31,1>,mm>& hog,
const long cell_draw_size = 15
const long cell_draw_size = 15,
const float min_response_threshold = 0.0
)
{
// make sure requires clause is not broken
......@@ -1167,7 +1170,7 @@ namespace dlib
const float val = hog[r/cell_draw_size][c/cell_draw_size](d) +
hog[r/cell_draw_size][c/cell_draw_size](d+mbars.size()) +
hog[r/cell_draw_size][c/cell_draw_size](d+mbars.size()*2);
if (val > 0)
if (val > min_response_threshold)
{
set_subm(himg, r, c, cell_draw_size, cell_draw_size) += val*mbars[d%mbars.size()];
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment