clean code
This commit is contained in:
parent
296aa8854c
commit
b6081438fa
@ -232,122 +232,6 @@ int qangle6(float dfdx, float dfdy)
|
|||||||
return index;
|
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>
|
template< typename T>
|
||||||
struct Decimate {
|
struct Decimate {
|
||||||
int shrinkage;
|
int shrinkage;
|
||||||
@ -363,8 +247,6 @@ struct Decimate {
|
|||||||
CV_Assert(cols * shrinkage == in.cols);
|
CV_Assert(cols * shrinkage == in.cols);
|
||||||
CV_Assert(rows * shrinkage == in.rows);
|
CV_Assert(rows * shrinkage == in.rows);
|
||||||
|
|
||||||
std::cout << "type: " << out.type() << std::endl;
|
|
||||||
|
|
||||||
for (int outIdx_y = 0; outIdx_y < rows; ++outIdx_y)
|
for (int outIdx_y = 0; outIdx_y < rows; ++outIdx_y)
|
||||||
{
|
{
|
||||||
T* outPtr = out.ptr<T>(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
|
struct ChannelStorage
|
||||||
{
|
{
|
||||||
@ -405,6 +287,13 @@ struct ChannelStorage
|
|||||||
#if defined USE_REFERENCE_VALUES
|
#if defined USE_REFERENCE_VALUES
|
||||||
cv::FileStorage imgs("/home/kellan/testInts.xml", cv::FileStorage::READ);
|
cv::FileStorage imgs("/home/kellan/testInts.xml", cv::FileStorage::READ);
|
||||||
char buff[33];
|
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
|
#else
|
||||||
// add gauss
|
// add gauss
|
||||||
cv::Mat gauss;
|
cv::Mat gauss;
|
||||||
@ -483,22 +372,15 @@ struct ChannelStorage
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#endif
|
for(int i = 0; i < HOG_BINS; ++i)
|
||||||
for(int i = 0; i < 6; ++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);
|
decimate(hist[i], shrunk);
|
||||||
cv::integral(shrunk, sum, cv::noArray(), CV_32S);
|
cv::integral(shrunk, sum, cv::noArray(), CV_32S);
|
||||||
hog.push_back(sum);
|
hog.push_back(sum);
|
||||||
#endif
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#if !defined USE_REFERENCE_VALUES
|
|
||||||
hog.push_back(mag);
|
hog.push_back(mag);
|
||||||
hog.insert(hog.end(), luvs.begin(), luvs.end());
|
hog.insert(hog.end(), luvs.begin(), luvs.end());
|
||||||
CV_Assert(hog.size() == 10);
|
CV_Assert(hog.size() == 10);
|
||||||
|
Loading…
x
Reference in New Issue
Block a user