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);