pull/137/head
marina.kolpakova 12 years ago
parent 296aa8854c
commit b6081438fa
  1. 138
      modules/objdetect/src/softcascade.cpp

@ -232,122 +232,6 @@ int qangle6(float dfdx, float dfdy)
return index;
}
//ToDo
void calcHistBins(const cv::Mat& grey, cv::Mat& magIntegral, std::vector<cv::Mat>& histInts,
const int bins, int shrinkage)
{
static const float magnitudeScaling = 1.f / sqrt(2);
CV_Assert( grey.type() == CV_8U);
float scale = 1.f / shrinkage;
const int rows = grey.rows + 1;
const int cols = grey.cols + 1;
cv::Mat df_dx(grey.rows, grey.cols, CV_32F),
df_dy(grey.rows, grey.cols, CV_32F), mag, angle;
// cv::Sobel(grey, df_dx, CV_32F, 1, 0);
// cv::Sobel(grey, df_dy, CV_32F, 0, 1);
for (int y = 1; y < grey.rows -1; ++y)
{
float* dx = df_dx.ptr<float>(y);
float* dy = df_dy.ptr<float>(y);
const uchar* gr = grey.ptr<uchar>(y);
const uchar* gr_down = grey.ptr<uchar>(y - 1);
const uchar* gr_up = grey.ptr<uchar>(y + 1);
for (int x = 1; x < grey.cols - 1; ++x)
{
float dx_a = gr[x + 1];
float dx_b = gr[x - 1];
dx[x] = dx_a - dx_b;
float dy_a = gr_up[x];
float dy_b = gr_down[x];
dy[x] = dy_a - dy_b;
}
}
cv::cartToPolar(df_dx, df_dy, mag, angle, true);
mag *= magnitudeScaling;
cv::Mat saturatedMag(grey.rows, grey.cols, CV_8UC1);
for (int y = 0; y < grey.rows; ++y)
{
float* rm = mag.ptr<float>(y);
uchar* mg = saturatedMag.ptr<uchar>(y);
for (int x = 0; x < grey.cols; ++x)
{
mg[x] = cv::saturate_cast<uchar>(rm[x]);
}
}
mag = saturatedMag;
histInts.clear();
std::vector<cv::Mat> hist;
for (int bin = 0; bin < bins; ++bin)
{
hist.push_back(cv::Mat(rows, cols, CV_8UC1));
}
for (int h = 0; h < saturatedMag.rows; ++h)
{
uchar* magnitude = saturatedMag.ptr<uchar>(h);
float* dfdx = df_dx.ptr<float>(h);
float* dfdy = df_dy.ptr<float>(h);
for (int w = 0; w < saturatedMag.cols; ++w)
{
hist[ qangle6(dfdx[w], dfdy[w]) ].ptr<uchar>(h)[w] = magnitude[w];
}
}
angle /= 60;
// for (int h = 0; h < saturatedMag.rows; ++h)
// {
// uchar* magnitude = saturatedMag.ptr<uchar>(h);
// float* ang = angle.ptr<float>(h);
// for (int w = 0; w < saturatedMag.cols; ++w)
// {
// hist[ (int)ang[w] ].ptr<uchar>(h)[w] = magnitude[w];
// }
// }
char buffer[33];
for (int bin = 0; bin < bins; ++bin)
{
cv::Mat shrunk, sum;
cv::imshow(std::string("hist[bin]") + itoa(bin, buffer, 10), hist[bin]);
cv::resize(hist[bin], shrunk, cv::Size(), scale, scale, cv::INTER_AREA);
cv::imshow(std::string("shrunk") + itoa(bin, buffer, 10), shrunk);
cv::integral(shrunk, sum);
cv::imshow(std::string("sum") + itoa(bin, buffer, 10), sum);
histInts.push_back(sum);
// std::cout << shrunk << std::endl << std::endl;
}
cv::Mat shrMag;
cv::imshow("mag", mag);
cv::resize(mag, shrMag, cv::Size(), scale, scale, cv::INTER_AREA);
cv::FileStorage fs("/home/kellan/actualChannels.xml", cv::FileStorage::WRITE);
cv::imshow("shrunk_channel", shrMag);
fs << "shrunk_channel6" << shrMag;
// cv::imshow("shrMag", shrMag);
cv::integral(shrMag, magIntegral, mag.depth());
// cv::imshow("magIntegral", magIntegral);
histInts.push_back(magIntegral);
}
template< typename T>
struct Decimate {
int shrinkage;
@ -363,8 +247,6 @@ struct Decimate {
CV_Assert(cols * shrinkage == in.cols);
CV_Assert(rows * shrinkage == in.rows);
std::cout << "type: " << out.type() << std::endl;
for (int outIdx_y = 0; outIdx_y < rows; ++outIdx_y)
{
T* outPtr = out.ptr<T>(outIdx_y);
@ -387,7 +269,7 @@ struct Decimate {
};
#define USE_REFERENCE_VALUES
// #define USE_REFERENCE_VALUES
struct ChannelStorage
{
@ -405,6 +287,13 @@ struct ChannelStorage
#if defined USE_REFERENCE_VALUES
cv::FileStorage imgs("/home/kellan/testInts.xml", cv::FileStorage::READ);
char buff[33];
for(int i = 0; i < HOG_LUV_BINS; ++i)
{
cv::Mat channel;
imgs[std::string("channel") + itoa(i, buff, 10)] >> channel;
hog.push_back(channel);
}
#else
// add gauss
cv::Mat gauss;
@ -483,22 +372,15 @@ struct ChannelStorage
}
}
#endif
for(int i = 0; i < 6; ++i)
for(int i = 0; i < HOG_BINS; ++i)
{
cv::Mat channel, shrunk, sum;
cv::Mat shrunk, sum;
#if defined USE_REFERENCE_VALUES
imgs[std::string("channel") + itoa(i, buff, 10)] >> channel;
hog.push_back(channel);
#else
decimate(hist[i], shrunk);
cv::integral(shrunk, sum, cv::noArray(), CV_32S);
hog.push_back(sum);
#endif
}
#if !defined USE_REFERENCE_VALUES
hog.push_back(mag);
hog.insert(hog.end(), luvs.begin(), luvs.end());
CV_Assert(hog.size() == 10);

Loading…
Cancel
Save