diff options
| author | Cheng Chen <chengchen@google.com> | 2022-08-26 14:29:32 -0700 |
|---|---|---|
| committer | Cheng Chen <chengchen@google.com> | 2022-08-26 14:32:17 -0700 |
| commit | fd45d113807eb00fd5b9e58784e48e662a6797b9 (patch) | |
| tree | 4a715e5fbcfc7906372f7206fec542d0851a518b /vp9/encoder/vp9_ext_ratectrl.c | |
| parent | 4bfab03e812f32167259f1e84356f862eb71fd44 (diff) | |
| download | libvpx-fd45d113807eb00fd5b9e58784e48e662a6797b9.tar libvpx-fd45d113807eb00fd5b9e58784e48e662a6797b9.tar.gz libvpx-fd45d113807eb00fd5b9e58784e48e662a6797b9.tar.bz2 libvpx-fd45d113807eb00fd5b9e58784e48e662a6797b9.zip | |
L2E: Add gop size and ARF existence to frame info
Pass the encode frame info to external ml model, with the information
of gop size and whether alt ref is used.
Change-Id: I55be2d3de83d7182c1a1a174e44ead7e19045c9d
Diffstat (limited to 'vp9/encoder/vp9_ext_ratectrl.c')
| -rw-r--r-- | vp9/encoder/vp9_ext_ratectrl.c | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/vp9/encoder/vp9_ext_ratectrl.c b/vp9/encoder/vp9_ext_ratectrl.c index d5b60b02a..7e38cc524 100644 --- a/vp9/encoder/vp9_ext_ratectrl.c +++ b/vp9/encoder/vp9_ext_ratectrl.c @@ -137,7 +137,7 @@ static int extrc_get_frame_type(FRAME_UPDATE_TYPE update_type) { vpx_codec_err_t vp9_extrc_get_encodeframe_decision( EXT_RATECTRL *ext_ratectrl, int show_index, int coding_index, int gop_index, - FRAME_UPDATE_TYPE update_type, + FRAME_UPDATE_TYPE update_type, const int gop_size, const int use_alt_ref, RefCntBuffer *ref_frame_bufs[MAX_INTER_REF_FRAMES], int ref_frame_flags, vpx_rc_encodeframe_decision_t *encode_frame_decision) { if (ext_ratectrl == NULL) { @@ -150,6 +150,8 @@ vpx_codec_err_t vp9_extrc_get_encodeframe_decision( encode_frame_info.coding_index = coding_index; encode_frame_info.gop_index = gop_index; encode_frame_info.frame_type = extrc_get_frame_type(update_type); + encode_frame_info.gop_size = gop_size; + encode_frame_info.use_alt_ref = use_alt_ref; vp9_get_ref_frame_info(update_type, ref_frame_flags, ref_frame_bufs, encode_frame_info.ref_frame_coding_indexes, |