opencv把jpg圖片轉化成yuv數據_opencv把Mat轉換成yuv

sprintf(imagepath, "/mnt/md/test/mtcnn/images/%d.jpg", i);
printf("imagepath:%s\n", imagepath);
cv::Mat cv_img = cv::imread(imagepath, 3);
if (cv_img.empty())
{
    fprintf(stderr, "cv::imread %s failed\n", imagepath);
    return -1;
}
printf("cv_img.rows:%d\n", cv_img.rows);
printf("cv_img.cols:%d\n", cv_img.cols);

cv::Mat cv_yuv;
cv::cvtColor(cv_img, cv_yuv, COLOR_BGR2YUV_I420);
//注意一個問題，執行完上一行代碼之后，cv_yuv和cv_img的寬是一樣的，但是cv_yuv的高是cv_img高的1.5倍，所以這兩個的高是不一樣的，
//有一次項目中把這兩個的高誤認為一樣，結果把cv_yuv.rows傳給了另一個函數，導致邏輯不對了。
pFrame = (unsigned char *)malloc((cv_img.rows)*(cv_img.cols)*3/2);
if(NULL == pFrame)
{
    printf("malloc pFrame memory error\n");
}
memset(pFrame, 0,(cv_img.rows)*(cv_img.cols)*3/2);    
memcpy(pFrame, cv_yuv.data, (cv_img.rows)*(cv_img.cols)*3/2);//注意這里用的是cv_img的寬高，如果使用cv_yuv的寬高，那么就不用*3/2了。

//如果想得到YUV圖片，可以寫下來。
fp = fopen("./a.yuv", "wb+");
if(NULL == fp)
{
    printf("file open error\n");
    return -1;
}
fwrite(pFrame, 1, (cv_img.rows)*(cv_img.cols)*3/2, fp);