如何更好地线程代码以减少计算时间

时间:2017-10-06 15:32:08

标签: c++ multithreading opencv parallel-processing video-processing

我编写了一个opencv代码,用于读取视频,在每个帧中查找红色像素,如果红色像素数超过一定数量,则将该帧导出为png文件。代码运行良好,但我正在寻找进一步缩短计算时间的方法,因为视频长达4-5小时。我正在阅读使用parallel_pipeline的帖子,并想知道是否使用它会大大加快这个过程。根据我读到的内容,我似乎必须为每个主要任务分配一个线程(读取视频帧,使用inRange进行颜色检测/阈值处理,以及保存图像)。所以我的问题是:

1)与opencv的默认多线程相比,这会加快进程吗?

2)鉴于代码需要做什么,是否有比parallel_pipeline更合适的多线程方法?

我对这个话题相当新,所以非常感谢任何帮助!

    /**
 * @CheckMotionParallel
 * @Motion detection using color detection and image thresholding
 */

//opencv
#include "opencv2/imgcodecs.hpp"
#include "opencv2/imgproc.hpp"
#include "opencv2/videoio.hpp"
#include <opencv2/highgui.hpp>
#include <opencv2/video.hpp>
//C
#include <stdio.h>
//C++
#include <iostream>
#include <sstream>
#include "tbb/blocked_range.h"
#include "tbb/parallel_for.h"
#include "tbb/parallel_reduce.h"
#include "tbb/task_scheduler_init.h"
#include "tbb/mutex.h"
#include "tbb/tbb_thread.h"
#include "tbb/blocked_range2d.h"

using namespace cv;
using namespace std;
using namespace tbb;

void help();
void help()
{
    cout
    << "--------------------------------------------------------------------------" << endl
    << "Note for program CheckMotion" << endl
    << "CheckMotion does the following"  << endl
    << "1) It searches each frame in a video and looks for a specified range of colors in the frame"                                                                   << endl
    << "2) Pixels falling within the range will be converted to white while everything else is turned to black"                     << endl
    << "3) For each frame, the program gives: frame number/time stamp, total pixel count, and white pixel count"                                       << endl
    << "4) For frames whose white pixel count exceeds a threshold, it will export those frames as individial png files" << endl
    << "--------------------------------------------------------------------------" << endl
    << endl;
}

int64 startTime;

int NumThreads = task_scheduler_init::default_num_threads();

int main(int argc, char**)
{
    //Print out program note
    help();

    ///Part I: Read-in the video

    VideoCapture cap("/Users/chi/Desktop/Video analyses/testvideo4.mp4");

    //Error message if the video cannot be opened
    //Create an object denoting the frames
    //Create a window for showing the video as CheckMotion runs
    //For loop looking through frames

    if(cap.isOpened()) {

        startTime = getTickCount();

        Mat frame;
        for(;;)
        {
            //Show each frame in the video window previously created
            double tfreq = getTickFrequency();
            double secs = ((double) getTickCount()-startTime)/tfreq;

            cap >> frame;

            //            namedWindow("Frame");
            //            imshow("Frame",frame);
            //
            waitKey(10);
            //Create a string for frame number that gets updated for each cycle of the loop
            stringstream ss;
            ss << cap.get(CAP_PROP_POS_FRAMES);
            string FrameNumberString = ss.str();

            stringstream maskedfilename;
            stringstream rawfilename;
            //Create filenames for later use in result output and image save using frame number as ref
            maskedfilename << "/Users/chi/Desktop/test/masked" << FrameNumberString.c_str() << ".png";
            rawfilename << "/Users/chi/Desktop/test/raw" << FrameNumberString.c_str() << ".png";

            ///Part II: Image thresholding and image saving

            //Create an object representing new images after thresholding
            Mat masked;
            //inRange function that convert the pixels that fall within the specified range to white and everything else to black
            //The Range is specified by a lower [Scalar(200,200,200)] and an upper [Scalar(255,255,255)] threshold
            //A color is defined by its BGR score
            //The thresholded images will then be represented by the object "masked"
            inRange(frame, Scalar(10,0,90), Scalar(50,50,170), masked);

            //Creating integer variables for total pixel count and white pixel count for each frame
            int totalpixel;
            int whitepixel;

            //Total pixel count equals the number of rows and columns of the frame
            totalpixel = masked.rows*masked.cols;
            //Using countNonZero function to count the number of white pixels
            whitepixel = countNonZero(masked);
            //Output frame number, total pixel count and white pixel count for each frame

            //Exit the loop when reaching the last frame (i.e. pixel count drops to 0)
            if(totalpixel==0){
                cout << "End of the video" << endl;
                cout << "Number of threads: " << NumThreads << endl;
                cap.release();
                break;
            }

            else {
                cout
                << "Frame:" << ss.str() << endl
                << "Number of total pixels:" << totalpixel << endl
                << "Pixels of target colors:" << whitepixel << endl
                << "Run time = " << fixed << secs << "seconds" << endl
                << endl;
                //Save the frames with white pixel count larger than a user-determined value (100 in present case)
                //Save both the orignal as well as the procesed images
                if (whitepixel > 50){
                imwrite(rawfilename.str(),frame);
                imwrite(maskedfilename.str(),masked);
                }
            }
        }
    }
}

1 个答案:

答案 0 :(得分:1)

只需删除此行:)

waitKey(10);

然后将endl替换为'\n'