From 3147bcce8b12d1b1917dee117209b5425c6a576d Mon Sep 17 00:00:00 2001
From: Jim Martens <github@2martens.de>
Date: Sun, 7 May 2017 16:13:54 +0200
Subject: [PATCH] [CCV] Added ability to visualize feature maps

Signed-off-by: Jim Martens <github@2martens>
---
 ccv/sheet4/lab_pyramid.cpp | 16 ++++++++++++++++
 ccv/sheet4/lab_pyramid.h   |  6 ++++++
 2 files changed, 22 insertions(+)

diff --git a/ccv/sheet4/lab_pyramid.cpp b/ccv/sheet4/lab_pyramid.cpp
index caac4b5..16b5b97 100644
--- a/ccv/sheet4/lab_pyramid.cpp
+++ b/ccv/sheet4/lab_pyramid.cpp
@@ -133,3 +133,19 @@ void lab_pyramid::visualize_dog() {
     cv::waitKey(0);
   }
 }
+
+void lab_pyramid::visualize_feature_maps() {
+  cv::namedWindow("CS F L");
+  cv::imshow("CS F L", _cs_F_l);
+  cv::namedWindow("CS F L");
+  cv::imshow("SC F L", _sc_F_l);
+  cv::namedWindow("CS F A");
+  cv::imshow("CS F A", _cs_F_a);
+  cv::namedWindow("SC F A");
+  cv::imshow("SC F A", _sc_F_a);
+  cv::namedWindow("CS F B");
+  cv::imshow("CS F B", _cs_F_b);
+  cv::namedWindow("CS F B");
+  cv::imshow("SC F B", _sc_F_b);
+  cv::waitKey(0);
+}
diff --git a/ccv/sheet4/lab_pyramid.h b/ccv/sheet4/lab_pyramid.h
index fbc51bd..da2a92c 100644
--- a/ccv/sheet4/lab_pyramid.h
+++ b/ccv/sheet4/lab_pyramid.h
@@ -85,6 +85,12 @@ public:
      * Has to be called after compute_dog.
      */
     void static compute_feature_maps();
+
+    /**
+     * Visualizes the feature maps.
+     * Has to be called after compute_feature_maps.
+     */
+    void static visualize_feature_maps();
 };