Opencv C++ Problem No.5 HSV色彩空间变换

题目源：https://github.com/gzr2017/ImageProcessing100Wen

题目描述： 

 

 参考答案：

//注意，很多强制转换是容易错的地方
#include <iostream>
#include <opencv2/opencv.hpp>
using namespace std;
using namespace cv;
Mat bgr2hsv(Mat img){//right
    int cols(img.cols),rows(img.rows);
    Mat hsv=Mat::zeros(rows,cols,CV_32FC3);
    for(int i=0;i<rows;i++){
        for(int j=0;j<cols;j++){
            float b=(float)img.at<Vec3b>(i,j)[0]/255;
            float g=(float)img.at<Vec3b>(i,j)[1]/255;
            float r=(float)img.at<Vec3b>(i,j)[2]/255;
            float max=fmax(b,fmax(g,r));
            float min=fmin(b,fmin(g,r));
            float h,s,v;
            if(min==max){
                h=0;
            }else if(min==b){
                h=60*(g-r)/(max-min)+60;
            }else if(min==r){
                h=60*(b-g)/(max-min)+180;
            }else if(min==g){
                h=60*(r-b)/(max-min)+300;
            }
            s=max-min;
            v=max;
            hsv.at<Vec3f>(i,j)[0]=h;
            hsv.at<Vec3f>(i,j)[1]=s;
            hsv.at<Vec3f>(i,j)[2]=v;
        }
    }
    return hsv;
}
Mat hsv2bgr(Mat hsv){
    int rows(hsv.rows),cols(hsv.cols);
    Mat bgr(Mat::zeros(rows,cols,CV_8UC3));
    for(int i=0;i<rows;i++){
        for(int j=0;j<cols;j++){
            double c=(double)hsv.at<Vec3f>(i,j)[1];
            double _h=(double)hsv.at<Vec3f>(i,j)[0]/60;
            double x=(double)c*(1-abs(fmod(_h,2)-1));
            double v=(double)hsv.at<Vec3f>(i,j)[2];
            double r,g,b;
            if(_h<1 && _h>=0){
                r=v;
                g=v-c+x;
                b=v-c;
            }else if(_h<2){
                r=v-c+x;
                g=v;
                b=v-c;
            }else if(_h<3){
                r=v-c;
                g=v;
                b=v-c+x;
            }else if(_h<4){
                r=v-c;
                g=v-c+x;
                b=v;
            }else if(_h<5){
                r=v-c+x;
                g=v-c;
                b=v;
            }else if(_h<6){
                r=v;
                g=v-c;
                b=v-c+x;
            }
            bgr.at<Vec3b>(i,j)[0]=(uchar)(b*255);//这里的b*255也要打上括号，容易忽略
            bgr.at<Vec3b>(i,j)[1]=(uchar)(g*255);
            bgr.at<Vec3b>(i,j)[2]=(uchar)(r*255);
        }
    }
    return bgr;
}

Mat h180(Mat hsv){//right
    int rows(hsv.rows),cols(hsv.cols);
    for(int i=0;i<rows;i++){
        for(int j=0;j<cols;j++){
            hsv.at<Vec3f>(i,j)[0]=fmod(hsv.at<Vec3f>(i,j)[0]+180,360);//看定义，得fmod()
            //fmod(hsv.at<Vec3f>(y, x)[0] + 180, 360)
        }
    }
    return hsv;
}

int main(int argc, const char* argv[]){
    // read image
    Mat img = cv::imread("C:/Users/79490/Desktop/ImageProcessing100Wen-master/ImageProcessing100Wen-master/Question_01_10/imori.jpg",cv::IMREAD_COLOR);

    // BGR -> HSV
    Mat hsv = bgr2hsv(img);

    // Inverse Hue
    hsv = h180(hsv);

    // Gray -> Binary
    Mat out = hsv2bgr(hsv);

    imshow("sample", out);
    waitKey(0);
    destroyAllWindows();

    return 0;
}

 这里，如果用Opencv的cvtColor函数可能导致错误，通过实验我发现按照题目里的定义生成的hsv图片与cvtColor()数生成的hsv图片不一样，具体可查看Opencv源码中对于COLOR_BGR2HSV的定义。