C++代码

  1. // file name: convert.cpp
  2. #include <opencv2/opencv.hpp>
  4. // BGR 转 YUV
  5. void BGR2YUV(const cv::Mat bgrImg, cv::Mat &y, cv::Mat &u, cv::Mat &v)
  6. {
  7.     cv::Mat out;
  8.     cv::cvtColor(bgrImg, out, cv::COLOR_BGR2YUV);
  9.     cv::bgr channel[3];
  10.     cv::split(out, channel);
  12.     y = channel[0];
  13.     u = channel[1];
  14.     v = channel[2];
  15. }
  17. // YUV 转 BGR
  18. void YUV2BGR(const cv::Mat y, const cv::Mat u, const cv::Mat v, cv::Mat& bgrImg)
  19. {
  20.     std::vector<cv::Mat> inChannels;
  21.     inChannels.push_back(y);
  22.     inChannels.push_back(u);
  23.     inChannels.push_back(v);
  25.     // 合并3个单独的 channel 进一个矩阵
  26.     cv::Mat yuvImg;
  27.     cv::merge(inChannels, yuvImg);
  29.     cv::cvtColor(yuvImg, bgrImg, cv::COLOR_YUV2BGR);
  30. }
  32. // 使用例子
  33. int main()
  34. {
  35.     cv::Mat origImg = cv::imread("test.png");
  37.     cv::Mat y, u, v;
  38.     BGR2YUV(origImg, y, u, v);
  40.     cv::Mat bgrImg;
  41.     YUV2BGR(y, u, v, bgrImg);
  43.     cv::imshow("origImg", origImg);
  44.     cv::imshow("Y channel", y);
  45.     cv::imshow("U channel", u);
  46.     cv::imshow("V channel", v);
  47.     cv::imshow("converted bgrImg", bgrImg);
  48.     cv::waitKey(0);
  50.     return 0;
  51. }

编译

  1. g++ -o convert convert.cpp  --std=c++11 `pkg-config --cflags --libs opencv`
  2. ./convert

Python代码

  1. # file name: convert.py
  2. import cv2
  4. def bgr2yuv(img):
  5.     yuv_img = cv2.cvtColor(img, cv2.COLOR_BGR2YUV)
  6.     y, u, v = cv2.split(yuv_img)
  8.     return y, u, v
  10. def yuv2bgr(y, u, v):
  11.     yuv_img = cv2.merge([y, u, v])
  12.     bgr_img = cv2.cvtColor(yuv_img, cv2.COLOR_YUV2BGR)
  14.     return bgr_img
  16. def main():
  17.     orig_img = cv2.imread('test.png')
  19.     y, u, v = bgr2yuv(orig_img)
  21.     bgr_img = yuv2bgr(y, u, v)
  23.     cv2.imshow('orig_img', orig_img)
  24.     cv2.imshow('Y channel', y)
  25.     cv2.imshow('U channel', u)
  26.     cv2.imshow('V channel', v)
  27.     cv2.imshow('bgr_img', bgr_img)
  28.     cv2.waitKey(0)
  30. if __name__ == '__main__':
  31.     main()

运行程序：

  1. python3 convert.py

https://vra.github.io/2019/12/19/opencv-bgr-yuv-conversion/#more