Update demo.cpp

Added HTTP stream over multi-client MJPEG  sever, added ESC Key quit and resizable cv window

demo/demo/demo.cpp

@@ -3,6 +3,7 @@
 #include <stdlib.h> /* srand, rand */
 #include <unistd.h>
 #include <mutex>
+#include <https_stream.h> //https_stream
 
 #include "CenternetDetection.h"
 #include "MobilenetDetection.h"
@@ -11,17 +12,18 @@
 bool gRun;
 bool SAVE_RESULT = false;
 
-void sig_handler(int signo) {
+void sig_handler(int signo)
+{
     std::cout << "request gateway stop\n";
     gRun = false;
 }
 
-int main(int argc, char *argv[]) {
+int main(int argc, char *argv[])
+{
 
     std::cout << "detection\n";
     signal(SIGINT, sig_handler);
 
-
     std::string net = "yolo3_berkeley.rt";
     if (argc > 1)
         net = argv[1];
@@ -40,6 +42,9 @@ int main(int argc, char *argv[]) {
     bool show = true;
     if (argc > 6)
         show = atoi(argv[6]);
+    int port = 0;
+    if (argc > 7)
+        port = atoi(argv[7]);
 
     if (n_batch < 1 || n_batch > 64)
         FatalError("Batch dim not supported");
@@ -80,7 +85,8 @@ int main(int argc, char *argv[]) {
         std::cout << "camera started\n";
 
     cv::VideoWriter resultVideo;
-    if(SAVE_RESULT) {
+    if (SAVE_RESULT)
+    {
         int w = cap.get(cv::CAP_PROP_FRAME_WIDTH);
         int h = cap.get(cv::CAP_PROP_FRAME_HEIGHT);
         resultVideo.open("result.mp4", cv::VideoWriter::fourcc('M', 'P', '4', 'V'), 30, cv::Size(w, h));
@@ -89,15 +95,19 @@ int main(int argc, char *argv[]) {
     cv::Mat frame;
     if (show)
         cv::namedWindow("detection", cv::WINDOW_NORMAL);
+    cv::moveWindow("detection", 0, 0);
+    cv::resizeWindow("detection", 1352, 1013);
 
     std::vector<cv::Mat> batch_frame;
     std::vector<cv::Mat> batch_dnn_input;
 
-    while(gRun) {
+    while (gRun)
+    {
         batch_dnn_input.clear();
         batch_frame.clear();
 
-        for(int bi=0; bi< n_batch; ++bi){
+        for (int bi = 0; bi < n_batch; ++bi)
+        {
             cap >> frame;
             if (!frame.data)
                 break;
@@ -114,14 +124,25 @@ int main(int argc, char *argv[]) {
         detNN->update(batch_dnn_input, n_batch);
         detNN->draw(batch_frame);
 
-        if(show){
-            for(int bi=0; bi< n_batch; ++bi){
+        if (show)
+        {
+            for (int bi = 0; bi < n_batch; ++bi)
+            {
                 cv::imshow("detection", batch_frame[bi]);
-                cv::waitKey(1);
+                
             }
         }
+        if (cv::waitKey(1) == 27)
+        {
+            break;
+        }
         if (n_batch == 1 && SAVE_RESULT)
             resultVideo << frame;
+       
+        if (port > 0)
+        {
+            send_mjpeg(batch_frame[0], port, 400000, 40);
+        }
     }
 
     std::cout << "detection end\n";
@@ -130,10 +151,11 @@ int main(int argc, char *argv[]) {
     std::cout << COL_GREENB << "\n\nTime stats:\n";
     std::cout << "Min: " << *std::min_element(detNN->stats.begin(), detNN->stats.end()) / n_batch << " ms\n";
     std::cout << "Max: " << *std::max_element(detNN->stats.begin(), detNN->stats.end()) / n_batch << " ms\n";
-    for(int i=0; i<detNN->stats.size(); i++) mean += detNN->stats[i]; mean /= detNN->stats.size();
-    std::cout<<"Avg: "<<mean/n_batch<<" ms\t"<<1000/(mean/n_batch)<<" FPS\n"<<COL_END;   
-    
+    for (int i = 0; i < detNN->stats.size(); i++)
+        mean += detNN->stats[i];
+    mean /= detNN->stats.size();
+    std::cout << "Avg: " << mean / n_batch << " ms\t" << 1000 / (mean / n_batch) << " FPS\n"
+              << COL_END;
 
     return 0;
 }
-