package/mpp
1
0
Fork 0
mirror of https://github.com/nyanmisaka/mpp.git synced 2025-10-12 20:40:07 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat[test]: Add PSNR info for video encoder

Browse Source 
Note that the PSNR derived from SSE may not equals to PSNR from video
analyser. Because SSE returned from hardware is consist of both luma and
chroma samples. And SSE of chroma samples multiplied by some weight.
But the difference is within 1db, usually no more than 0.5db.

Change-Id: I294d473c656f39ce98dd0768cd6945ee38e89b14
Signed-off-by: Johnson Ding <johnson.ding@rock-chips.com>
This commit is contained in:
Johnson Ding
2024-12-20 15:40:21 +08:00
committed by Herman Chen
parent 52699a26ab
commit 16ab9b0d4d
1 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
test/mpi_enc_test.c
View File

@@ -21,6 +21,7 @@
#define MODULE_TAG "mpi_enc_test"
#include <string.h>
#include <math.h>
#include "rk_mpi.h"
#include "mpp_env.h"
@@ -658,6 +659,8 @@ MPP_RET test_mpp_run(MpiEncMultiCtxInfo *info)
RK_U32 cap_num = 0;
DataCrc checkcrc;
MPP_RET ret = MPP_OK;
RK_FLOAT psnr_const = 0;
RK_U32 sse_unit_in_pixel = 0;
memset(&checkcrc, 0, sizeof(checkcrc));
checkcrc.sum = mpp_malloc(RK_ULONG, 512);
@@ -689,6 +692,10 @@ MPP_RET test_mpp_run(MpiEncMultiCtxInfo *info)
}
mpp_packet_deinit(&packet);
sse_unit_in_pixel = p->type == MPP_VIDEO_CodingAVC ? 16 : 8;
psnr_const = (16 + log2(MPP_ALIGN(p->width, sse_unit_in_pixel) *
MPP_ALIGN(p->height, sse_unit_in_pixel)));
}
while (!p->pkt_eos) {
MppMeta meta = NULL;
@@ -919,6 +926,8 @@ MPP_RET test_mpp_run(MpiEncMultiCtxInfo *info)
RK_S32 lt_idx = -1;
RK_S32 avg_qp = -1, bps_rt = -1;
RK_S32 use_lt_idx = -1;
RK_S64 sse = 0;
RK_FLOAT psnr = 0;
if (MPP_OK == mpp_meta_get_s32(meta, KEY_TEMPORAL_ID, &temporal_id))
log_len += snprintf(log_buf + log_len, log_size - log_len,
@@ -938,6 +947,12 @@ MPP_RET test_mpp_run(MpiEncMultiCtxInfo *info)
if (MPP_OK == mpp_meta_get_s32(meta, KEY_ENC_USE_LTR, &use_lt_idx))
log_len += snprintf(log_buf + log_len, log_size - log_len, " vi");
if (MPP_OK == mpp_meta_get_s64(meta, KEY_ENC_SSE, &sse)) {
psnr = 3.01029996 * (psnr_const - log2(sse));
log_len += snprintf(log_buf + log_len, log_size - log_len,
" psnr %.4f", psnr);
}
}
mpp_log_q(quiet, "chn %d %s\n", chn, log_buf);