implemented KCF fracker class and included opencv in cmake

2026-03-02 08:53:52 -08:00 · 2020-06-26 10:29:08 -03:00
parent 517f2896d1
commit ae48651660
5 changed files with 262 additions and 2 deletions
--- a/src/examples/Example_opencv.cpp
+++ b/src/examples/Example_opencv.cpp
@@ -0,0 +1,143 @@
+/**
+ * @file
+ * @brief Source file for Example Executable (example app for libopenshot)
+ * @author Jonathan Thomas <jonathan@openshot.org>
+ *
+ * @ref License
+ */
+
+/* LICENSE
+ *
+ * Copyright (c) 2008-2019 OpenShot Studios, LLC
+ * <http://www.openshotstudios.com/>. This file is part of
+ * OpenShot Library (libopenshot), an open-source project dedicated to
+ * delivering high quality video editing and animation solutions to the
+ * world. For more information visit <http://www.openshot.org/>.
+ *
+ * OpenShot Library (libopenshot) is free software: you can redistribute it
+ * and/or modify it under the terms of the GNU Lesser General Public License
+ * as published by the Free Software Foundation, either version 3 of the
+ * License, or (at your option) any later version.
+ *
+ * OpenShot Library (libopenshot) is distributed in the hope that it will be
+ * useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with OpenShot Library. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include <fstream>
+#include <iostream>
+#include <memory>
+#include <opencv2/opencv.hpp>
+#include "../../include/CVTracker.h"
+
+
+#include "../../include/OpenShot.h"
+#include "../../include/CrashHandler.h"
+
+using namespace openshot;
+using namespace cv;
+
+
+cv::Mat qimage2mat( std::shared_ptr<QImage>& qimage) {
+
+    cv::Mat mat = cv::Mat(qimage->height(), qimage->width(), CV_8UC4, (uchar*)qimage->bits(), qimage->bytesPerLine());
+    cv::Mat mat2 = cv::Mat(mat.rows, mat.cols, CV_8UC3 );
+    int from_to[] = { 0,0,  1,1,  2,2 };
+    cv::mixChannels( &mat, 1, &mat2, 1, from_to, 3 );
+    return mat2.clone();
+};
+
+
+int main(int argc, char* argv[]) {
+
+    openshot::Settings *s = openshot::Settings::Instance();
+    s->HARDWARE_DECODER = 2; // 1 VA-API, 2 NVDEC, 6 VDPAU
+    s->HW_DE_DEVICE_SET = 0;
+
+    std::string input_filepath = TEST_MEDIA_PATH;
+    input_filepath += "Boneyard Memories.mp4";
+
+    openshot::FFmpegReader r9(input_filepath);
+    r9.Open();
+    r9.DisplayInfo();
+
+    /* WRITER ---------------- */
+    openshot::FFmpegWriter w9("metadata.mp4");
+
+    // Set options
+    w9.SetAudioOptions(true, "libmp3lame", r9.info.sample_rate, r9.info.channels, r9.info.channel_layout, 128000);
+    w9.SetVideoOptions(true, "libx264", r9.info.fps, 1024, 576, openshot::Fraction(1,1), false, false, 3000000);
+
+    w9.info.metadata["title"] = "testtest";
+	w9.info.metadata["artist"] = "aaa";
+	w9.info.metadata["album"] = "bbb";
+	w9.info.metadata["year"] = "2015";
+	w9.info.metadata["description"] = "ddd";
+	w9.info.metadata["comment"] = "eee";
+	w9.info.metadata["comment"] = "comment";
+    w9.info.metadata["copyright"] = "copyright OpenShot!";
+
+    // Open writer
+    w9.Open();
+    // opencv display window
+    cv::namedWindow("Display Image", cv::WINDOW_NORMAL );
+
+    CVTracker kcfTracker;
+    bool trackerInit = false;
+
+
+    for (long int frame = 1100; frame <= 1500; frame++)
+    {
+        //int frame_number = (rand() % 750) + 1;
+        int frame_number = frame;
+        std::shared_ptr<openshot::Frame> f = r9.GetFrame(frame_number);
+
+        std::shared_ptr<QImage> qimage = f->GetImage();
+        // qimage->convertToFormat(QImage::Format_RGB888);
+        
+        // convert to opencv image
+        cv::Mat cvimage = qimage2mat(qimage);
+        cvtColor(cvimage, cvimage, CV_RGB2BGR);
+
+
+        if(!trackerInit){
+            // Rect2d bbox(287, 23, 86, 320);
+            Rect2d bbox = selectROI("Display Image", cvimage);
+
+            kcfTracker.initTracker(bbox, cvimage);
+            trackerInit = true;
+        }
+        else{
+            trackerInit = kcfTracker.trackFrame(cvimage);
+
+        }
+
+
+        // opencv code
+        if ( !cvimage.data )
+        {
+            std::cout << "No image data \n";
+        }
+        
+        cv::imshow("Display Image", cvimage);
+
+        cv::waitKey(30);
+
+
+        w9.WriteFrame(f);
+    }
+
+    // Close writer & reader
+    w9.Close();
+
+    // Close timeline
+    r9.Close();
+
+	std::cout << "Completed successfully!" << std::endl;
+
+    return 0;
+}