计算两段yuv格式视频流中每一帧的psnr值

最新推荐文章于 2025-05-30 09:14:51 发布

算法工程师000001

最新推荐文章于 2025-05-30 09:14:51 发布

阅读量4.2k

点赞数

CC 4.0 BY-SA版权

分类专栏：虚拟视点文章标签：视频流批处理视频 psnr

本文链接：https://blog.youkuaiyun.com/u012361418/article/details/47043699

现在虚拟视点生成了yuv格式的视频流，如何计算每一帧的psnr值，进行了如下步骤：

1.对yuv视频流进行转换为jpg图片；

2.把jpg图片转化为avi视频流，便于批处理；

3.对avi视频中的每一帧计算psnr值。

程序：

#include <stdlib.h>
#include <stdio.h>
#include <math.h>
#include <cv.h>
#include <highgui.h>
#define NUM_FRAME 100 //只处理前100帧，根据视频帧数可修改
void CalcPsnr(const char *in1,const char *in2)
{
	cv::VideoCapture vc1;
	cv::VideoCapture vc2;
	bool flag1 = vc1.open(in1);
	bool flag2 = vc2.open(in2);
	if (!flag1||!flag2)
	{
		printf("avi file open error \n");
		system("pause");
		exit(-1);
	}

	int frmCount1 = vc1.get(CV_CAP_PROP_FRAME_COUNT);
	int frmCount2 = vc2.get(CV_CAP_PROP_FRAME_COUNT);
	printf("frmCount: %d \n", frmCount1);
	printf("frmCount: %d \n", frmCount2);
	for (int i = 0; i < frmCount1; i++)
	{
		printf("%d/%d \n", i + 1, frmCount1);
		cv::Mat image_ref;
		vc1 >> image_ref;
		cv::Mat image_obj;
		vc2 >> image_obj;
		double mse = 0;
		double div_r =