opencv/samples/gpu/performance/performance.cpp

84 lines
2.0 KiB
C++
Raw Normal View History

#include <iomanip>
#include "performance.h"
using namespace std;
using namespace cv;
void Test::gen(Mat& mat, int rows, int cols, int type)
{
mat.create(rows, cols, type);
Mat mat8u(rows, cols * mat.elemSize(), CV_8U, mat.data, mat.step);
RNG rng(0);
rng.fill(mat, RNG::UNIFORM, Scalar(0), Scalar(256));
}
void TestSystem::run()
{
cout << setiosflags(ios_base::left);
cout << " " << setw(10) << "CPU, ms" << setw(10) << "GPU, ms"
<< setw(10) << "SPEEDUP" << "DESCRIPTION\n";
cout << resetiosflags(ios_base::left);
vector<Test*>::iterator it = tests_.begin();
for (; it != tests_.end(); ++it)
{
Test* test = *it;
cout << endl << test->name() << ":\n";
test->run();
flush();
}
cout << setiosflags(ios_base::fixed | ios_base::left);
2011-01-25 09:32:45 +01:00
cout << "\nAverage GPU Speedup: x" << setprecision(3)
<< speedup_total_ / num_subtests_called_ << endl;
cout << resetiosflags(ios_base::fixed | ios_base::left);
}
void TestSystem::flush()
{
if (!can_flush_)
return;
int cpu_time = static_cast<int>(cpu_elapsed_ / getTickFrequency() * 1000.0);
int gpu_time = static_cast<int>(gpu_elapsed_ / getTickFrequency() * 1000.0);
2011-01-25 09:32:45 +01:00
2011-01-24 11:39:28 +01:00
double speedup = static_cast<double>(cpu_time) / gpu_time;
2011-01-25 09:32:45 +01:00
speedup_total_ += speedup;
cpu_elapsed_ = 0;
gpu_elapsed_ = 0;
cout << " " << setiosflags(ios_base::fixed | ios_base::left);
stringstream stream;
stream << cpu_time;
cout << setw(10) << stream.str();
stream.str("");
stream << gpu_time;
cout << setw(10) << stream.str();
stream.str("");
stream << "x" << setprecision(3) << speedup;
cout << setw(10) << stream.str();
cout << description_.str();
description_.str("");
cout << resetiosflags(ios_base::fixed | ios_base::left) << endl;
2011-01-24 11:56:11 +01:00
can_flush_ = false;
2011-01-25 09:32:45 +01:00
num_subtests_called_++;
}
int main()
{
TestSystem::instance()->run();
return 0;
}