mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-12 20:42:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fixes #12692: Update samples/gpu/videoreader.cpp

Browse Source 
Change iteration to iterate separately when the video is decode on a GPU to account for different number of frames
This commit is contained in:
Sam Radhakrishnan 2018-09-30 00:36:20 +05:30 committed by Sam Radhakrishnan
parent ea92c2b1a6
commit 699db77293
1 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
samples/gpu/video_reader.cpp
View File

@ -35,6 +35,8 @@ int main(int argc, const char* argv[])
std::vector<double> cpu_times; std::vector<double> cpu_times;
std::vector<double> gpu_times; std::vector<double> gpu_times;
int gpu_frame_count=0, cpu_frame_count=0;
for (;;) for (;;)
{ {
tm.reset(); tm.start(); tm.reset(); tm.start();
@ -42,14 +44,23 @@ int main(int argc, const char* argv[])
break; break;
tm.stop(); tm.stop();
cpu_times.push_back(tm.getTimeMilli()); cpu_times.push_back(tm.getTimeMilli());
cpu_frame_count++;
cv::imshow("CPU", frame);
if (cv::waitKey(3) > 0)
break;
}
for (;;)
{
tm.reset(); tm.start(); tm.reset(); tm.start();
if (!d_reader->nextFrame(d_frame)) if (!d_reader->nextFrame(d_frame))
break; break;
tm.stop(); tm.stop();
gpu_times.push_back(tm.getTimeMilli()); gpu_times.push_back(tm.getTimeMilli());
gpu_frame_count++;
cv::imshow("CPU", frame);
cv::imshow("GPU", d_frame); cv::imshow("GPU", d_frame);
if (cv::waitKey(3) > 0) if (cv::waitKey(3) > 0)
@ -66,8 +77,8 @@ int main(int argc, const char* argv[])
double cpu_avg = std::accumulate(cpu_times.begin(), cpu_times.end(), 0.0) / cpu_times.size(); double cpu_avg = std::accumulate(cpu_times.begin(), cpu_times.end(), 0.0) / cpu_times.size();
double gpu_avg = std::accumulate(gpu_times.begin(), gpu_times.end(), 0.0) / gpu_times.size(); double gpu_avg = std::accumulate(gpu_times.begin(), gpu_times.end(), 0.0) / gpu_times.size();
std::cout << "CPU : Avg : " << cpu_avg << " ms FPS : " << 1000.0 / cpu_avg << std::endl; std::cout << "CPU : Avg : " << cpu_avg << " ms FPS : " << 1000.0 / cpu_avg << " Frames " << cpu_frame_count << std::endl;
std::cout << "GPU : Avg : " << gpu_avg << " ms FPS : " << 1000.0 / gpu_avg << std::endl; std::cout << "GPU : Avg : " << gpu_avg << " ms FPS : " << 1000.0 / gpu_avg << " Frames " << gpu_frame_count << std::endl;
} }
return 0; return 0;