diff options
| author | Vivek R <123vivekr@gmail.com> | 2020-07-10 19:31:13 +0530 |
|---|---|---|
| committer | GStreamer Marge Bot <gitlab-merge-bot@gstreamer-foundation.org> | 2021-09-22 20:19:10 +0000 |
| commit | d5e94da30ca54495b50d5db7bef5cd34e251d754 (patch) | |
| tree | 3d9f794aaeb7b4beb939219d7dd038f448e4e081 | |
| parent | d28f11998deadc8036bb63aa5bdc80beb1a9fbe4 (diff) | |
| download | gstreamer-plugins-bad-d5e94da30ca54495b50d5db7bef5cd34e251d754.tar.gz | |
opencv: cvtracker: add draw property
This property controls the drawing of rectangle around the tracked object.
Part-of: <https://gitlab.freedesktop.org/gstreamer/gst-plugins-bad/-/merge_requests/2454>
| -rw-r--r-- | ext/opencv/gstcvtracker.cpp | 15 | ||||
| -rw-r--r-- | ext/opencv/gstcvtracker.h | 1 |
2 files changed, 16 insertions, 0 deletions
diff --git a/ext/opencv/gstcvtracker.cpp b/ext/opencv/gstcvtracker.cpp index 829d9a2c6..492c9cf9b 100644 --- a/ext/opencv/gstcvtracker.cpp +++ b/ext/opencv/gstcvtracker.cpp @@ -78,6 +78,7 @@ enum PROP_INITIAL_WIDTH, PROP_INITIAL_HEIGHT, PROP_ALGORITHM, + PROP_DRAW, }; #define GST_OPENCV_TRACKER_ALGORITHM (tracker_algorithm_get_type ()) @@ -207,6 +208,11 @@ gst_cvtracker_class_init (GstCVTrackerClass * klass) GST_OPENCV_TRACKER_ALGORITHM_MEDIANFLOW, (GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS))); + g_object_class_install_property (gobject_class, PROP_DRAW, + g_param_spec_boolean ("draw-rect", "Display", + "Draw rectangle around tracked object", + TRUE, (GParamFlags) G_PARAM_READWRITE)); + gst_element_class_set_static_metadata (element_class, "cvtracker", "Filter/Effect/Video", @@ -233,6 +239,7 @@ gst_cvtracker_init (GstCVTracker * filter) #else filter->tracker = cv::TrackerMedianFlow::create(); #endif + filter->draw = TRUE; filter->post_debug_info = TRUE; gst_opencv_video_filter_set_in_place (GST_OPENCV_VIDEO_FILTER_CAST (filter), @@ -262,6 +269,9 @@ gst_cvtracker_set_property (GObject * object, guint prop_id, case PROP_ALGORITHM: filter->algorithm = g_value_get_enum (value); break; + case PROP_DRAW: + filter->draw = g_value_get_boolean (value); + break; default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; @@ -338,6 +348,9 @@ gst_cvtracker_get_property (GObject * object, guint prop_id, case PROP_ALGORITHM: g_value_set_enum (value, filter->algorithm); break; + case PROP_DRAW: + g_value_set_boolean (value, filter->draw); + break; default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; @@ -382,6 +395,8 @@ gst_cvtracker_transform_ip (GstOpencvVideoFilter * base, filter->roi->x, filter->roi->y, filter->roi->width, filter->roi->height); gst_element_post_message (GST_ELEMENT (filter), msg); + if (filter->draw) + cv::rectangle (img, *filter->roi, cv::Scalar (255, 0, 0), 2, 1); if (!(filter->post_debug_info)) filter->post_debug_info = TRUE; } else if (filter->post_debug_info) { diff --git a/ext/opencv/gstcvtracker.h b/ext/opencv/gstcvtracker.h index b817c86b8..d3531efbc 100644 --- a/ext/opencv/gstcvtracker.h +++ b/ext/opencv/gstcvtracker.h @@ -80,6 +80,7 @@ struct _GstCVTracker guint width; guint height; gint algorithm; + gboolean draw; gboolean post_debug_info; cv::Ptr<cv::Tracker> tracker; |