Merge pull request #1815 from luxonis/fix/rvc2_poe_tests

Fix/rvc2 poe tests

Author: aljazdu

examples/cpp/Benchmark/benchmark_nn.cpp

@@ -1,8 +1,19 @@
+#include <atomic>
+#include <csignal>
 #include <depthai/depthai.hpp>
 #include <iostream>
 #include <opencv2/opencv.hpp>
 
+std::atomic<bool> quitEvent(false);
+
+void signalHandler(int) {
+    quitEvent = true;
+}
+
 int main() {
+    signal(SIGTERM, signalHandler);
+    signal(SIGINT, signalHandler);
+
     // First prepare the model for benchmarking
     std::shared_ptr<dai::Device> device = std::make_shared<dai::Device>();
 
@@ -50,10 +61,13 @@ int main() {
     pipeline.start();
     inputQueue->send(inputFrame);
 
-    while(pipeline.isRunning()) {
+    while(pipeline.isRunning() && !quitEvent) {
         auto benchmarkReport = outputQueue->get<dai::BenchmarkReport>();
         std::cout << "FPS is " << benchmarkReport->fps << std::endl;
     }
 
+    pipeline.stop();
+    pipeline.wait();
+
     return 0;
 }

examples/cpp/NeuralNetwork/neural_network.cpp

@@ -1,9 +1,20 @@
+#include <atomic>
+#include <csignal>
 #include <iostream>
 #include <opencv2/opencv.hpp>
 
 #include "depthai/depthai.hpp"
 
+std::atomic<bool> quitEvent(false);
+
+void signalHandler(int) {
+    quitEvent = true;
+}
+
 int main() {
+    signal(SIGTERM, signalHandler);
+    signal(SIGINT, signalHandler);
+
     // Create pipeline
     dai::Pipeline pipeline;
 
@@ -29,10 +40,14 @@ int main() {
     pipeline.start();
 
     // Main loop
-    while(pipeline.isRunning()) {
+    while(pipeline.isRunning() && !quitEvent) {
         auto inNNData = qNNData->get<dai::NNData>();
         auto tensor = inNNData->getFirstTensor<float>();
         std::cout << "Received NN data: " << tensor.shape()[0] << "x" << tensor.shape()[1] << std::endl;
     }
+
+    pipeline.stop();
+    pipeline.wait();
+
     return 0;
 }

examples/cpp/NeuralNetwork/neural_network_multi_input.cpp

@@ -1,11 +1,22 @@
+#include <atomic>
+#include <csignal>
 #include <iostream>
 #include <opencv2/opencv.hpp>
 #include <xtensor/containers/xarray.hpp>
 
 #include "depthai/depthai.hpp"
 #include "depthai/modelzoo/Zoo.hpp"
 
+std::atomic<bool> quitEvent(false);
+
+void signalHandler(int) {
+    quitEvent = true;
+}
+
 int main() {
+    signal(SIGTERM, signalHandler);
+    signal(SIGINT, signalHandler);
+
     // Decode the image using OpenCV
     cv::Mat lenaImage = cv::imread(LENNA_PATH);
     cv::resize(lenaImage, lenaImage, cv::Size(256, 256));
@@ -48,7 +59,7 @@ int main() {
     lennaInputQueue->send(daiLenaImage);
 
     // Main loop
-    while(pipeline.isRunning()) {
+    while(pipeline.isRunning() && !quitEvent) {
         auto inNNData = qNNData->get<dai::NNData>();
         auto tensor = inNNData->getFirstTensor<float>();
         auto tensor_uint8 = xt::eval(xt::squeeze(xt::cast<uint8_t>(tensor), 0));
@@ -68,5 +79,8 @@ int main() {
         }
     }
 
+    pipeline.stop();
+    pipeline.wait();
+
     return 0;
 }

examples/cpp/NeuralNetwork/neural_network_multi_input_combined.cpp

@@ -1,12 +1,22 @@
-#include <iostream>
+#include <atomic>
+#include <csignal>
 #include <opencv2/opencv.hpp>
 #include <xtensor/containers/xadapt.hpp>
 #include <xtensor/containers/xarray.hpp>
 
 #include "depthai/depthai.hpp"
 #include "depthai/modelzoo/Zoo.hpp"
 
+std::atomic<bool> quitEvent(false);
+
+void signalHandler(int) {
+    quitEvent = true;
+}
+
 int main() {
+    signal(SIGTERM, signalHandler);
+    signal(SIGINT, signalHandler);
+
     // Decode the image using OpenCV
     cv::Mat lenaImageCv = cv::imread(LENNA_PATH);
     cv::resize(lenaImageCv, lenaImageCv, cv::Size(256, 256));
@@ -51,7 +61,7 @@ int main() {
     pipeline.start();
 
     // Main loop
-    while(pipeline.isRunning()) {
+    while(pipeline.isRunning() && !quitEvent) {
         nnDataInputQueue->send(inputNNData);
         auto inNNData = qNNData->get<dai::NNData>();
         auto tensor = inNNData->getFirstTensor<float>();
@@ -72,5 +82,8 @@ int main() {
         }
     }
 
+    pipeline.stop();
+    pipeline.wait();
+
     return 0;
 }