Method CaptureThread

async_detector/AsyncDetector.cpp:253–378

\brief AsyncDetector::CaptureThread \param fileName \param framesQue \param stopFlag

/// \param stopFlag
///
void AsyncDetector::CaptureThread(std::string fileName, int startFrame, int* framesCount, float* fps, FramesQueue* framesQue, bool* stopFlag)
{
    cv::VideoCapture capture;
    if (fileName.size() == 1)
        capture.open(atoi(fileName.c_str()));
    else
        capture.open(fileName);

    if (!capture.isOpened())
    {
        *stopFlag = true;
        std::cerr << "Can't open " << fileName << std::endl;
        return;
    }
    *framesCount = cvRound(capture.get(cv::CAP_PROP_FRAME_COUNT));
    capture.set(cv::CAP_PROP_POS_FRAMES, startFrame);

    time_point_t startTimeStamp = std::chrono::system_clock::now();

    *fps = std::max(1.f, (float)capture.get(cv::CAP_PROP_FPS));
	int frameHeight = cvRound(capture.get(cv::CAP_PROP_FRAME_HEIGHT));

    // Detector
    config_t detectorConfig;

#ifdef _WIN32
    std::string pathToModel = "../../data/";
#else
    std::string pathToModel = "../data/";
#endif

#if 0
	detectorConfig.emplace("modelConfiguration", pathToModel + "yolov3-tiny.cfg");
	detectorConfig.emplace("modelBinary", pathToModel + "yolov3-tiny.weights");
#else
    detectorConfig.emplace("modelConfiguration", pathToModel + "yolov4-csp.cfg");
    detectorConfig.emplace("modelBinary", pathToModel + "yolov4-csp.weights");
#endif
    detectorConfig.emplace("classNames", pathToModel + "coco.names");
    detectorConfig.emplace("confidenceThreshold", "0.3");
    detectorConfig.emplace("maxCropRatio", "3.0");
	
#if 1
    detectorConfig.emplace("white_list", "person");
    detectorConfig.emplace("white_list", "car");
    detectorConfig.emplace("white_list", "bicycle");
    detectorConfig.emplace("white_list", "motorbike");
    detectorConfig.emplace("white_list", "bus");
    detectorConfig.emplace("white_list", "truck");
#endif

    // Tracker
    const int minStaticTime = 5;

    TrackerSettings trackerSettings;
	trackerSettings.SetDistance(tracking::DistCenters);
    trackerSettings.m_kalmanType = tracking::KalmanLinear;
    trackerSettings.m_filterGoal = tracking::FilterRect;