Method trackClip

src/CVTracker.cpp:70–124

Track object in the whole clip or in a given interval

// Track object in the whole clip or in a given interval
void CVTracker::trackClip(openshot::Clip& video,
                          size_t _start,
                          size_t _end,
                          bool process_interval)
{
    video.Open();
    if (!json_interval) {
        start = _start; end = _end;
        if (!process_interval || end <= 1 || end - start == 0) {
            start = int(video.Start() * video.Reader()->info.fps.ToFloat()) + 1;
            end   = int(video.End()   * video.Reader()->info.fps.ToFloat()) + 1;
        }
    } else {
        start = int(start + video.Start() * video.Reader()->info.fps.ToFloat()) + 1;
        end   = int(video.End()   * video.Reader()->info.fps.ToFloat()) + 1;
    }
    if (error) return;
    processingController->SetError(false, "");

    bool trackerInit = false;
    lostCount = 0;  // reset lost counter once at the start

    for (size_t frame = start; frame <= end; ++frame) {
        if (processingController->ShouldStop()) return;

        auto f      = video.GetFrame(frame);
        cv::Mat img = f->GetImageCV();

        if (frame == start) {
            bbox = cv::Rect2d(
                int(bbox.x      * img.cols),
                int(bbox.y      * img.rows),
                int(bbox.width  * img.cols),
                int(bbox.height * img.rows)
            );
        }

        if (!trackerInit) {
            initTracker(img, frame);
            trackerInit = true;
            lostCount    = 0;
        }
        else {
            // trackFrame now manages lostCount and will re-init internally
            trackFrame(img, frame);

            // record whatever bbox we have now
            FrameData fd = GetTrackedData(frame);
        }

        processingController->SetProgress(
            uint(100 * (frame - start) / (end - start))
        );
    }
}

// Initialize the tracker
bool CVTracker::initTracker(cv::Mat &frame, size_t frameId)