fix: convert to single channel image before feeding into Canny (Kuren…

…to#1) Co-authored-by: Chen Wei <[email protected]>
naevatec · Nov 12, 2020 · 66c0418 · 66c0418
1 parent 5952590
commit 66c0418
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/gst-plugins/kmssamplefilter.cpp b/src/gst-plugins/kmssamplefilter.cpp
@@ -218,8 +218,9 @@ kms_sample_filter_transform_frame_ip (GstVideoFilter *filter,
 
   if (filter_type == KMS_SAMPLE_FILTER_TYPE_EDGES) {
     GST_DEBUG ( "Calculating edges");
-    Canny ( (*sample_filter->priv->cv_image), output_image,
-            edge_threshold, 255);
+    Mat gray_image;
+    cvtColor ( (*sample_filter->priv->cv_image), gray_image, COLOR_BGR2GRAY);
+    Canny (gray_image, output_image, edge_threshold, 255);
   } else if (filter_type == KMS_SAMPLE_FILTER_TYPE_GREY) {
     GST_DEBUG ( "Calculating black&white image");
     cvtColor ( (*sample_filter->priv->cv_image), output_image, COLOR_BGR2GRAY );