1 sprintf(imagepath, "/mnt/md/test/mtcnn/images/%d.jpg", i);
4 printf("imagepath:%s\n", imagepath);
5 cv::Mat cv_img = cv::imread(imagepath, 3);
6 if (cv_img.empty())
7 {
8 fprintf(stderr, "cv::imread %s failed\n", imagepath);
9 return -1;
10 }
11 printf("cv_img.rows:%d\n", cv_img.rows);
12 printf("cv_img.cols:%d\n", cv_img.cols);
13
14 cv::Mat cv_yuv;
15 cv::cvtColor(cv_img, cv_yuv, COLOR_BGR2YUV_I420);
//注意一个问题,执行完上一行代码之后,cv_yuv和cv_img的宽是一样的,但是cv_yuv的高是cv_img高的1.5倍,所以这两个的高是不一样的,
//有一次项目中把这两个的高误认为一样,结果把cv_yuv.rows传给了另一个函数,导致逻辑不对了。
16 pFrame = (unsigned char *)malloc((cv_img.rows)*(cv_img.cols)*3/2);
17 if(NULL == pFrame)
18 {
19 printf("malloc pFrame memory error\n");
20 }
21 memset(pFrame, 0,(cv_img.rows)*(cv_img.cols)*3/2);
22 memcpy(pFrame, cv_yuv.data, (cv_img.rows)*(cv_img.cols)*3/2);//注意这里用的是cv_img的宽高,如果使用cv_yuv的宽高,那么就不用*3/2了。
23
24 //如果想得到YUV图片,可以写下来。
25 fp = fopen("./a.yuv", "wb+");
26 if(NULL == fp)
27 {
28 printf("file open error\n");
29 return -1;
30 }
31 fwrite(pFrame, 1, (cv_img.rows)*(cv_img.cols)*3/2, fp);