FaSTTUBe/GST-Tensordecoder-ov_ep
37
0
Fork 0
Code Issues Pull Requests 1 Actions Packages 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_WIDTH,
PROP_INITIAL_HEIGHT, PROP_INITIAL_HEIGHT,
PROP_ALGORITHM, PROP_ALGORITHM,
PROP_DRAW,
}; };
#define GST_OPENCV_TRACKER_ALGORITHM (tracker_algorithm_get_type ()) #define GST_OPENCV_TRACKER_ALGORITHM (tracker_algorithm_get_type ())
@ -207,6 +208,11 @@ gst_cvtracker_class_init (GstCVTrackerClass * klass)
GST_OPENCV_TRACKER_ALGORITHM_MEDIANFLOW, GST_OPENCV_TRACKER_ALGORITHM_MEDIANFLOW,
(GParamFlags) (G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS))); (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, gst_element_class_set_static_metadata (element_class,
"cvtracker", "cvtracker",
"Filter/Effect/Video", "Filter/Effect/Video",
@ -233,6 +239,7 @@ gst_cvtracker_init (GstCVTracker * filter)
#else #else
filter->tracker = cv::TrackerMedianFlow::create(); filter->tracker = cv::TrackerMedianFlow::create();
#endif #endif
filter->draw = TRUE;
filter->post_debug_info = TRUE; filter->post_debug_info = TRUE;
gst_opencv_video_filter_set_in_place (GST_OPENCV_VIDEO_FILTER_CAST (filter), 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: case PROP_ALGORITHM:
filter->algorithm = g_value_get_enum (value); filter->algorithm = g_value_get_enum (value);
break; break;
case PROP_DRAW:
filter->draw = g_value_get_boolean (value);
break;
default: default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break; break;
@ -338,6 +348,9 @@ gst_cvtracker_get_property (GObject * object, guint prop_id,
case PROP_ALGORITHM: case PROP_ALGORITHM:
g_value_set_enum (value, filter->algorithm); g_value_set_enum (value, filter->algorithm);
break; break;
case PROP_DRAW:
g_value_set_boolean (value, filter->draw);
break;
default: default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
break; break;
@ -382,6 +395,8 @@ gst_cvtracker_transform_ip (GstOpencvVideoFilter * base,
filter->roi->x, filter->roi->y, filter->roi->width, filter->roi->x, filter->roi->y, filter->roi->width,
filter->roi->height); filter->roi->height);
gst_element_post_message (GST_ELEMENT (filter), msg); 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)) if (!(filter->post_debug_info))
filter->post_debug_info = TRUE; filter->post_debug_info = TRUE;
} else if (filter->post_debug_info) { } else if (filter->post_debug_info) {

1
ext/opencv/gstcvtracker.h
View File

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