diff options
| -rw-r--r-- | tracker-pt/ftnoir_tracker_pt_dialog.cpp | 1 | ||||
| -rw-r--r-- | tracker-pt/module/point_extractor.cpp | 5 | ||||
| -rw-r--r-- | tracker-pt/pt-settings.hpp | 1 |
3 files changed, 7 insertions, 0 deletions
diff --git a/tracker-pt/ftnoir_tracker_pt_dialog.cpp b/tracker-pt/ftnoir_tracker_pt_dialog.cpp index 01ae83c9..1f6233f1 100644 --- a/tracker-pt/ftnoir_tracker_pt_dialog.cpp +++ b/tracker-pt/ftnoir_tracker_pt_dialog.cpp @@ -94,6 +94,7 @@ TrackerDialog_PT::TrackerDialog_PT(const QString& module_name) : pt_color_average, pt_color_natural, pt_color_red_only, + pt_color_green_only, pt_color_blue_only, }; diff --git a/tracker-pt/module/point_extractor.cpp b/tracker-pt/module/point_extractor.cpp index 00248029..cd6cea7c 100644 --- a/tracker-pt/module/point_extractor.cpp +++ b/tracker-pt/module/point_extractor.cpp @@ -128,6 +128,11 @@ void PointExtractor::color_to_grayscale(const cv::Mat& frame, cv::Mat1b& output) { switch (s.blob_color) { + case pt_color_green_only: + { + extract_single_channel(frame, 1, output); + break; + } case pt_color_blue_only: { extract_single_channel(frame, 0, output); diff --git a/tracker-pt/pt-settings.hpp b/tracker-pt/pt-settings.hpp index 62297713..9c4b6438 100644 --- a/tracker-pt/pt-settings.hpp +++ b/tracker-pt/pt-settings.hpp @@ -12,6 +12,7 @@ enum pt_color_type pt_color_red_only = 3, pt_color_average = 5, pt_color_blue_only = 6, + pt_color_green_only = 7, }; namespace pt_settings_detail { |