mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2024-06-02 13:20:40 +00:00
Code Issues 1 Projects Releases Wiki Activity

opencv: cvtracker: add draw property

Browse source 
This property controls the drawing of rectangle around the tracked object.

Part-of: <https://gitlab.freedesktop.org/gstreamer/gst-plugins-bad/-/merge_requests/2454>
This commit is contained in:
Vivek R 2020-07-10 19:31:13 +05:30 committed by GStreamer Marge Bot
parent d28f11998d
commit d5e94da30c
2 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
ext/opencv/gstcvtracker.cpp
View file

@ -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) {

1
ext/opencv/gstcvtracker.h
View file

@ -80,6 +80,7 @@ struct _GstCVTracker
guint width;
guint height;
gint algorithm;
gboolean draw;
gboolean post_debug_info;
cv::Ptr<cv::Tracker> tracker;