[TUTO][FIX] Fixing error when image is RGBa instead of BGR

lagadic · Oct 8, 2024 · 307082c · 307082c
1 parent 5436980
commit 307082c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tutorial/detection/dnn/tutorial-dnn-object-detection-live.cpp b/tutorial/detection/dnn/tutorial-dnn-object-detection-live.cpp
@@ -338,6 +338,11 @@ int main(int argc, const char *argv[])
     cv::Mat frame;
     while (true) {
       capture >> frame;
+      if (frame.type() == CV_8UC4) {
+        // RGBa format is not supported by the class, converting to BGR format
+        cv::Mat cpy = frame;
+        cv::cvtColor(cpy, frame, cv::COLOR_RGBA2BGR);
+      }
       if (frame.empty())
         break;