[x265] [PATCH] rc: use a sliding window to calculate moving avg SatdCost for ABR Reset logic
Deepthi Nandakumar
deepthi at multicorewareinc.com
Mon Oct 6 05:03:02 CEST 2014
Thanks, better now. Pushed this. Lets think about a frame rate dependent
sliding window a little more. A larger window will still catch black
frames, I think (after all, we did pretty decently with an "infinite"
window until now).
On Sun, Oct 5, 2014 at 9:56 PM, <aarthi at multicorewareinc.com> wrote:
> # HG changeset patch
> # User Aarthi Thirumalai
> # Date 1412460518 -19800
> # Sun Oct 05 03:38:38 2014 +0530
> # Node ID 577a5647d4424da206aab07bb002ed73349bcf4a
> # Parent b6d49505b179cb509aa76f3a065192f0b4926579
> rc: use a sliding window to calculate moving avg SatdCost for ABR Reset
> logic
>
> contains improvements for detection of scene changes within Rate Control
> to stabilize qp and prevent
> extreme spikes in bitrate. Removes the blockiness or distortions in the
> frames that a streak of low-detailed
> frames and prevents vbv from overreacting at the points of scene cuts.
>
> diff -r b6d49505b179 -r 577a5647d442 source/encoder/ratecontrol.cpp
> --- a/source/encoder/ratecontrol.cpp Thu Oct 02 16:47:55 2014 -0500
> +++ b/source/encoder/ratecontrol.cpp Sun Oct 05 03:38:38 2014 +0530
> @@ -40,6 +40,7 @@
> /* Amortize the partial cost of I frames over the next N frames */
> const double RateControl::s_amortizeFraction = 0.85;
> const int RateControl::s_amortizeFrames = 75;
> +const int RateControl::s_slidingWindowFrames = 20;
> const char *RateControl::s_defaultStatFileName = "x265_2pass.log";
>
> namespace {
> @@ -472,6 +473,12 @@
> m_framesDone = 0;
> m_residualCost = 0;
> m_partialResidualCost = 0;
> + for (int i = 0; i < s_slidingWindowFrames; i++)
> + {
> + m_satdCostWindow[i] = 0;
> + m_encodedBitsWindow[i] = 0;
> + }
> + m_sliderPos = 0;
>
> /* 720p videos seem to be a good cutoff for cplxrSum */
> double tuneCplxFactor = (m_param->rc.cuTree && m_ncu > 3600) ? 2.5 :
> 1;
> @@ -1344,6 +1351,26 @@
> g_sliceTypeToChar[m_sliceType],
> g_sliceTypeToChar[rce->sliceType]);
> }
> }
> + else
> + {
> + if (m_isAbr)
> + {
> + double slidingWindowCplxSum = 0;
> + int start = m_sliderPos > s_slidingWindowFrames ?
> m_sliderPos : 0;
> + for (int cnt = 0; cnt < s_slidingWindowFrames; cnt++, start++)
> + {
> + int pos = start % s_slidingWindowFrames;
> + slidingWindowCplxSum *= 0.5;
> + if (!m_satdCostWindow[pos])
> + break;
> + slidingWindowCplxSum += m_satdCostWindow[pos] /
> (CLIP_DURATION(m_frameDuration) / BASE_FRAME_DURATION);
> + }
> + rce->movingAvgSum = slidingWindowCplxSum;
> + m_satdCostWindow[m_sliderPos % s_slidingWindowFrames] =
> rce->lastSatd;
> + m_sliderPos++;
> + }
> + }
> +
> if (m_sliceType == B_SLICE)
> {
> /* B-frames don't have independent rate control, but rather get
> the
> @@ -1478,7 +1505,7 @@
> * tolerances, the bit distribution approaches that of 2pass.
> */
>
> double wantedBits, overflow = 1;
> - rce->movingAvgSum = m_shortTermCplxSum;
> +
> m_shortTermCplxSum *= 0.5;
> m_shortTermCplxCount *= 0.5;
> m_shortTermCplxSum += m_currentSatd /
> (CLIP_DURATION(m_frameDuration) / BASE_FRAME_DURATION);
> @@ -1619,8 +1646,12 @@
> {
> if (!m_isAbrReset && rce->movingAvgSum > 0)
> {
> + int64_t shrtTermWantedBits = (int64_t) (X265_MIN(m_sliderPos,
> s_slidingWindowFrames) * m_bitrate * m_frameDuration);
> + int64_t shrtTermTotalBitsSum = 0;
> // Reset ABR if prev frames are blank to prevent further
> sudden overflows/ high bit rate spikes.
> - double underflow = (m_totalBits - m_wantedBitsWindow) /
> abrBuffer;
> + for (int i = 0; i < s_slidingWindowFrames ; i++)
> + shrtTermTotalBitsSum += m_encodedBitsWindow[i];
> + double underflow = (shrtTermTotalBitsSum -
> shrtTermWantedBits) / abrBuffer;
> const double epsilon = 0.0001f;
> if (underflow < epsilon && !isFrameDone)
> {
> @@ -1792,9 +1823,10 @@
> q = X265_MAX(q0 / 2, q);
> }
>
> - if (!m_isCbr)
> + if (!m_isCbr || m_isAbrReset)
> q = X265_MAX(q0, q);
>
> +
> if (m_rateFactorMaxIncrement)
> {
> double qpNoVbv = x265_qScale2qp(q0);
> @@ -2111,30 +2143,29 @@
> }
> }
> }
> - if (!m_isAbrReset)
> +
> + if (m_param->rc.aqMode || m_isVbv)
> {
> - if (m_param->rc.aqMode || m_isVbv)
> + if (pic->m_qpaRc)
> {
> - if (pic->m_qpaRc)
> - {
> - for (uint32_t i = 0; i < pic->getFrameHeightInCU(); i++)
> - pic->m_avgQpRc += pic->m_qpaRc[i];
> + for (uint32_t i = 0; i < pic->getFrameHeightInCU(); i++)
> + pic->m_avgQpRc += pic->m_qpaRc[i];
>
> - pic->m_avgQpRc /= (pic->getFrameHeightInCU() *
> pic->getFrameWidthInCU());
> - rce->qpaRc = pic->m_avgQpRc;
> - // copy avg RC qp to m_avgQpAq. To print out the correct
> qp when aq/cutree is disabled.
> - pic->m_avgQpAq = pic->m_avgQpRc;
> - }
> + pic->m_avgQpRc /= (pic->getFrameHeightInCU() *
> pic->getFrameWidthInCU());
> + rce->qpaRc = pic->m_avgQpRc;
> + // copy avg RC qp to m_avgQpAq. To print out the correct qp
> when aq/cutree is disabled.
> + pic->m_avgQpAq = pic->m_avgQpRc;
> + }
>
> - if (pic->m_qpaAq)
> - {
> - for (uint32_t i = 0; i < pic->getFrameHeightInCU(); i++)
> - pic->m_avgQpAq += pic->m_qpaAq[i];
> + if (pic->m_qpaAq)
> + {
> + for (uint32_t i = 0; i < pic->getFrameHeightInCU(); i++)
> + pic->m_avgQpAq += pic->m_qpaAq[i];
>
> - pic->m_avgQpAq /= (pic->getFrameHeightInCU() *
> pic->getFrameWidthInCU());
> - }
> + pic->m_avgQpAq /= (pic->getFrameHeightInCU() *
> pic->getFrameWidthInCU());
> }
> }
> +
> // Write frame stats into the stats file if 2 pass is enabled.
> if (m_param->rc.bStatWrite)
> {
> @@ -2199,6 +2230,9 @@
> }
> m_wantedBitsWindow += m_frameDuration * m_bitrate;
> m_totalBits += bits - rce->rowTotalBits;
> + int pos = m_sliderPos - m_param->frameNumThreads;
> + if (pos >= 0)
> + m_encodedBitsWindow[pos % s_slidingWindowFrames] = actualBits;
> }
>
> if (m_2pass)
> diff -r b6d49505b179 -r 577a5647d442 source/encoder/ratecontrol.h
> --- a/source/encoder/ratecontrol.h Thu Oct 02 16:47:55 2014 -0500
> +++ b/source/encoder/ratecontrol.h Sun Oct 05 03:38:38 2014 +0530
> @@ -173,7 +173,9 @@
> int64_t m_totalBits; /* total bits used for already encoded
> frames */
> int m_framesDone; /* # of frames passed through RateCotrol
> already */
> double m_fps;
> -
> + int64_t m_satdCostWindow[50];
> + int m_sliderPos;
> + int64_t m_encodedBitsWindow[50];
> /* a common variable on which rateControlStart, rateControlEnd and
> rateControUpdateStats waits to
> * sync the calls to these functions. For example
> * -F2:
> @@ -233,6 +235,7 @@
>
> static const double s_amortizeFraction;
> static const int s_amortizeFrames;
> + static const int s_slidingWindowFrames;
> static const char *s_defaultStatFileName;
>
> int m_residualFrames;
> _______________________________________________
> x265-devel mailing list
> x265-devel at videolan.org
> https://mailman.videolan.org/listinfo/x265-devel
>
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mailman.videolan.org/pipermail/x265-devel/attachments/20141006/4e8d3603/attachment.html>
More information about the x265-devel
mailing list