Colormap

xddd-ys · Jul 20, 2015 · a7bbb9e · a7bbb9e
1 parent ff51ae8
commit a7bbb9e
Show file tree

Hide file tree

Showing 5 changed files with 200 additions and 0 deletions.
diff --git a/Colormap/colormap.cpp b/Colormap/colormap.cpp
@@ -0,0 +1,64 @@
+/**
+ * OpenCV Colormap  Example
+ *
+ * Copyright 2015 by Satya Mallick <[email protected]>
+ *
+ */
+
+#include "opencv2/opencv.hpp"
+
+using namespace cv;
+using namespace std;
+
+string colormap_name(int id)
+{
+    switch(id){
+        case COLORMAP_AUTUMN :
+            return "COLORMAP_AUTUMN";
+        case COLORMAP_BONE :
+            return "COLORMAP_BONE";
+        case COLORMAP_JET :
+            return "COLORMAP_JET";
+        case COLORMAP_WINTER :
+            return "COLORMAP_WINTER";
+        case COLORMAP_RAINBOW :
+            return "COLORMAP_RAINBOW";
+        case COLORMAP_OCEAN :
+            return "COLORMAP_OCEAN";
+        case COLORMAP_SUMMER:
+            return "COLORMAP_SUMMER";
+        case COLORMAP_SPRING :
+            return "COLORMAP_SPRING";
+        case COLORMAP_COOL :
+            return "COLORMAP_COOL";
+        case COLORMAP_HSV :
+            return "COLORMAP_HSV";
+        case COLORMAP_PINK :
+            return "COLORMAP_PINK";
+        case COLORMAP_HOT :
+            return "COLORMAP_HOT";
+
+    }
+
+    return "NONE";
+}
+
+int main( int argc, char** argv )
+{
+    // Read 8-bit grayscale image
+    Mat im = imread("pluto.jpg", IMREAD_GRAYSCALE);
+
+    Mat im_out = Mat::zeros(600, 800, CV_8UC3);
+
+    for (int i=0; i < 4; i++){
+        for(int j=0; j < 3; j++){
+            int k = i + j * 4;
+            Mat im_color = im_out(Rect(i * 200, j * 200, 200, 200));
+            applyColorMap(im, im_color, k);
+            putText(im_color, colormap_name(k), Point(30, 180), CV_FONT_HERSHEY_DUPLEX, 0.5, Scalar::all(255), 1, CV_AA);
+        }
+    }
+
+    imshow("Pseudo Colored", im_out);
+    waitKey(0);
+}
diff --git a/Colormap/colormap.py b/Colormap/colormap.py
@@ -0,0 +1,51 @@
+#!/usr/bin/env python
+
+'''
+    OpenCV Colormap  Example
+    
+    Copyright 2015 by Satya Mallick <[email protected]>
+    
+'''
+
+
+import cv2
+import numpy as np
+
+
+def colormap_name(id) :
+    switcher = {
+        0 : "COLORMAP_AUTUMN",
+        1 : "COLORMAP_BONE",
+        2 : "COLORMAP_JET",
+        3 : "COLORMAP_WINTER",
+        4 : "COLORMAP_RAINBOW",
+        5 : "COLORMAP_OCEAN",
+        6 : "COLORMAP_SUMMER",
+        7 : "COLORMAP_SPRING",
+        8 : "COLORMAP_COOL",
+        9 : "COLORMAP_HSV",
+        10: "COLORMAP_PINK",
+        11: "COLORMAP_HOT"
+
+    }
+    return switcher.get(id, "NONE")
+
+
+if __name__ == '__main__' :
+
+    im = cv2.imread("pluto.jpg", cv2.IMREAD_GRAYSCALE)
+    im_out = np.zeros((600, 800, 3), np.uint8);
+
+    for i in xrange(0,4) :
+        for j in xrange(0,3) :
+            k = i + j * 4
+            im_color = cv2.applyColorMap(im, k)
+            cv2.putText(im_color, colormap_name(k), (30, 180), cv2.FONT_HERSHEY_DUPLEX, 0.5, (255, 255, 255), 1, cv2.CV_AA);
+
+            ix200 = i * 200
+            jx200 = j * 200
+
+            im_out[ jx200 : jx200 + 200 , ix200 : ix200 + 200 , : ] = im_color
+
+    cv2.imshow("Pseudo Colored", im_out);
+    cv2.waitKey(0);
diff --git a/Colormap/custom_colormap.cpp b/Colormap/custom_colormap.cpp
@@ -0,0 +1,47 @@
+/**
+ * OpenCV Custom Colormap  Example
+ *
+ * Copyright 2015 by Satya Mallick <[email protected]>
+ *
+ */
+
+
+#include "opencv2/opencv.hpp"
+
+using namespace cv;
+using namespace std;
+
+Mat applyCustomColorMap(Mat& im_gray)
+{
+    unsigned char b[] = {255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,253,251,249,247,245,242,241,238,237,235,233,231,229,227,225,223,221,219,217,215,213,211,209,207,205,203,201,199,197,195,193,191,189,187,185,183,181,179,177,175,173,171,169,167,165,163,161,159,157,155,153,151,149,147,145,143,141,138,136,134,132,131,129,126,125,122,121,118,116,115,113,111,109,107,105,102,100,98,97,94,93,91,89,87,84,83,81,79,77,75,73,70,68,66,64,63,61,59,57,54,52,51,49,47,44,42,40,39,37,34,33,31,29,27,25,22,20,18,17,14,13,11,9,6,4,2,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
+
+
+    unsigned char g[] = { 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,254,252,250,248,246,244,242,240,238,236,234,232,230,228,226,224,222,220,218,216,214,212,210,208,206,204,202,200,198,196,194,192,190,188,186,184,182,180,178,176,174,171,169,167,165,163,161,159,157,155,153,151,149,147,145,143,141,139,137,135,133,131,129,127,125,123,121,119,117,115,113,111,109,107,105,103,101,99,97,95,93,91,89,87,85,83,82,80,78,76,74,72,70,68,66,64,62,60,58,56,54,52,50,48,46,44,42,40,38,36,34,32,30,28,26,24,22,20,18,16,14,12,10,8,6,4,2,0 };
+
+
+    unsigned char r[] = {195,194,193,191,190,189,188,187,186,185,184,183,182,181,179,178,177,176,175,174,173,172,171,170,169,167,166,165,164,163,162,161,160,159,158,157,155,154,153,152,151,150,149,148,147,146,145,143,142,141,140,139,138,137,136,135,134,133,131,130,129,128,127,126,125,125,125,125,125,125,125,125,125,125,125,125,125,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126};
+
+    Mat channels[] = {Mat(256,1, CV_8U, b), Mat(256,1, CV_8U, g), Mat(256,1, CV_8U, r)};
+    Mat lut; // Create a lookup table
+    merge(channels, 3, lut);
+
+    Mat im_color;
+    LUT(im_gray, lut, im_color);
+
+    return im_color;
+
+}
+
+
+
+int main( int argc, char** argv )
+{
+    // Read 8-bit grayscale image
+    Mat im = imread("pluto.jpg", IMREAD_GRAYSCALE);
+    cvtColor(im.clone(), im, COLOR_GRAY2BGR);
+    Mat im_color = applyCustomColorMap(im);
+
+    imshow("Pseudo Colored Image", im_color);
+    waitKey(0);
+
+}
diff --git a/Colormap/custom_colormap.py b/Colormap/custom_colormap.py
@@ -0,0 +1,38 @@
+#!/usr/bin/env python
+
+'''
+    OpenCV Custom Colormap  Example
+    
+    Copyright 2015 by Satya Mallick <[email protected]>
+    
+'''
+
+import cv2
+import numpy as np
+
+def applyCustomColorMap(im_gray) :
+
+    lut = np.zeros((256, 1, 3), dtype=np.uint8)
+
+    lut[:, 0, 0] = [255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,253,251,249,247,245,242,241,238,237,235,233,231,229,227,225,223,221,219,217,215,213,211,209,207,205,203,201,199,197,195,193,191,189,187,185,183,181,179,177,175,173,171,169,167,165,163,161,159,157,155,153,151,149,147,145,143,141,138,136,134,132,131,129,126,125,122,121,118,116,115,113,111,109,107,105,102,100,98,97,94,93,91,89,87,84,83,81,79,77,75,73,70,68,66,64,63,61,59,57,54,52,51,49,47,44,42,40,39,37,34,33,31,29,27,25,22,20,18,17,14,13,11,9,6,4,2,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0]
+
+    lut[:, 0, 1] = [ 255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,255,254,252,250,248,246,244,242,240,238,236,234,232,230,228,226,224,222,220,218,216,214,212,210,208,206,204,202,200,198,196,194,192,190,188,186,184,182,180,178,176,174,171,169,167,165,163,161,159,157,155,153,151,149,147,145,143,141,139,137,135,133,131,129,127,125,123,121,119,117,115,113,111,109,107,105,103,101,99,97,95,93,91,89,87,85,83,82,80,78,76,74,72,70,68,66,64,62,60,58,56,54,52,50,48,46,44,42,40,38,36,34,32,30,28,26,24,22,20,18,16,14,12,10,8,6,4,2,0 ]
+
+
+    lut[:, 0, 2] = [195,194,193,191,190,189,188,187,186,185,184,183,182,181,179,178,177,176,175,174,173,172,171,170,169,167,166,165,164,163,162,161,160,159,158,157,155,154,153,152,151,150,149,148,147,146,145,143,142,141,140,139,138,137,136,135,134,133,131,130,129,128,127,126,125,125,125,125,125,125,125,125,125,125,125,125,125,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,127,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126,126]
+
+
+    im_color = cv2.LUT(im_gray, lut)
+
+    return im_color;
+
+
+if __name__  == '__main__' :
+
+    im = cv2.imread("pluto.jpg", cv2.IMREAD_GRAYSCALE);
+    im = cv2.cvtColor(im, cv2.COLOR_GRAY2BGR);
+    im_color = applyCustomColorMap(im);
+
+    cv2.imwrite('/tmp/colormap_algae.jpg', im_color)
+    cv2.imshow("Pseudo Colored Image", im_color);
+    cv2.waitKey(0);
diff --git a/Colormap/pluto.jpg b/Colormap/pluto.jpg