opencv-自定义色带产生类似jet的效果

/*!
*@brief  自定义伪彩图颜色
*/
Mat ColorMap(const Mat &img1, Mat &color) {
     
    //具体算法，主要思路是把灰度图对应的0～255的数值分别转换成：红、橙、黄、绿、青、蓝。
    //低值为蓝色高值为红色
    Mat img;
    img1.copyTo(img);
    uchar tmp2 = 0;
    Mat img_color(img.rows, img.cols, CV_8UC3);
#define IMG_B(img,y,x) img.at(y,x)[0]
#define IMG_G(img,y,x) img.at(y,x)[1]
#define IMG_R(img,y,x) img.at(y,x)[2]
    for(int y = 0; y < img.rows; y++) {
      
        for(int x = 0; x < img.cols; x++) {
     
            tmp2 = img.at<uchar>(y, x);
            if(tmp2 == 0) {
     
                IMG_B(img_color, y, x) = 0;
                IMG_G(img_color, y, x) = 0;
                IMG_R(img_color, y, x) = 0;
            } else if(tmp2 <= 51) {
     
                IMG_B(img_color, y, x) = 255;
                IMG_G(img_color, y, x) = tmp2 * 5;
                IMG_R(img_color, y, x) = 0;
            } else if(tmp2 <= 102) {
     
                tmp2 -= 51;
                IMG_B(img_color, y, x) = 255 - tmp2 * 5;
                IMG_G(img_color, y, x) = 255;
                IMG_R(img_color, y, x) = 0;
            } else if(tmp2 <= 153) {
     
                tmp2 -= 102;
                IMG_B(img_color, y, x) = 0;
                IMG_G(img_color, y, x) = 255;
                IMG_R(img_color, y, x) = tmp2 * 5;
            } else if(tmp2 <= 204) {
     
                tmp2 -= 153;
                IMG_B(img_color, y, x) = 0;
                IMG_G(img_color, y, x) = 255 - uchar(128.0 * tmp2 / 51.0 + 0.5);
                IMG_R(img_color, y, x) = 255;
            } else {
     
                tmp2 -= 204;
                IMG_B(img_color, y, x) = 0;
                IMG_G(img_color, y, x) = 127 - uchar(127.0 * tmp2 / 51.0 + 0.5);
                IMG_R(img_color, y, x) = 255;
            }
        }
    }
    img_color.copyTo(color);
    return color;
}