Server IP : 92.204.138.22 / Your IP : 3.15.1.36 Web Server : Apache System : Linux ns1009439.ip-92-204-138.us 4.18.0-553.8.1.el8_10.x86_64 #1 SMP Tue Jul 2 07:26:33 EDT 2024 x86_64 User : internationaljou ( 1019) PHP Version : 7.4.33 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : ON | Sudo : ON | Pkexec : ON Directory : /proc/self/root/proc/thread-self/root/usr/share/doc/svt-av1-libs/Docs/ |
Upload File : |
# Compound Mode Prediction Appendix ## 1. Description of the algorithm The general idea behind compound prediction is to generate a weighted average of two different predictions of the same block to develop a final prediction. Let Prediction\_1 and Prediction\_2 denote two different predictions of the same block. Sample p(i,j) in the compound prediction is then generated using sample p1(i,j) from Prediction\_1, sample p2(i,j) from Prediction\_2 and weight m(i,j) as follows: p(i,j) = m(i,j)p1(i,j) + (1-m(i,j))p2(i,j) Figure 1 illustrates the process of generating compound mode predictions. ![comp_mode_pred_fig1](./img/comp_mode_pred_fig1.png) ##### Figure 1. Compound mode prediction generation. Four different compound prediction types are supported: - Inter-Intra prediction: The mask (i.e. weights) are based on sample position relative to the block boundary. - Wedge prediction: The mask is based on a wedge codebook. Could be inter-inter or inter-intra prediction. - Distance-weighted compound prediction: The weights are based on the distance between the current frame and the reference frame. - Difference-weighted compound prediction: The weights are based on the difference between the two inter predictions. **Compound Inter-Intra Prediction** The compound inter-intra prediction mode is useful in blocks that contain previously occluded areas. Inter prediction is usually preferred for non-occluded content, whereas intra prediction is helpful in uncovered areas. A combined inter/intra prediction helps generate predictions for such cases that take advantage of the benefits of both inter prediction and intra prediction. Only H\_PRED, V\_PRED, DC\_PRED, and SMOOTH\_PRED intra modes are supported. The mask for the intra prediction P1(i, j) applies a smoothly decaying weight in the direction of intra prediction. The mask is inferred from a primitive 128-tap 1-D decaying function ```ii_weights1d(.)```. - ![math](http://latex.codecogs.com/gif.latex?m_{DC}(i,j)) = 0.5, - ![math](http://latex.codecogs.com/gif.latex?m_{HORZ}(i,j)) = ii\_weights1d(a\*j), - ![math](http://latex.codecogs.com/gif.latex?m_{VERT}(i,j)) = ii\_weights1d (a\*i), - ![math](http://latex.codecogs.com/gif.latex?m_{SMOOTH}(i,j)) = ii\_weights1d(a\*min(i, j)). *where a = 128/size\_of\_long\_edge(block\_size) and where* ![math](http://latex.codecogs.com/gif.latex?m_{DC})*,* ![math](http://latex.codecogs.com/gif.latex?m_{HORZ})*, *![math](http://latex.codecogs.com/gif.latex?m_{VERT})* and* ![math](http://latex.codecogs.com/gif.latex?m_{SMOOTH}) *are the masks for the inter-intra smooth modes involving the DC, horizontal, vertical and smooth intra prediction modes, respectively. The array* ii\_weights1d is given below ii_weights1d(.):\ 60, 58, 56, 54, 52, 50, 48, 47, 45, 44, 42, 41, 39, 38,\ 37, 35, 34, 33, 32, 31, 30, 29, 28, 27, 26, 25, 24, 23,\ 22, 22, 21, 20, 19, 19, 18, 18, 17, 16, 16, 15, 15, 14,\ 14, 13, 13, 12, 12, 12, 11, 11, 10, 10, 10, 9, 9, 9, 8,\ 8, 8, 8, 7, 7, 7, 7, 6, 6, 6, 6, 6, 5, 5, 5, 5, 5, 4,\ 4, 4, 4, 4, 4, 4, 4, 3, 3, 3, 3, 3, 3, 3, 3, 3, 2, 2,\ 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 1, 1, 1, 1, 1,\ 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1 **Compound Wedge Prediction** The general idea in compound wedge prediction is to generate a better prediction of areas around edges by combining two different predictions of the block. The feature makes use of a wedge codebook where wedge orientations are either horizontal, vertical or oblique with slopes: 2, -2, 0.5 and -0.5 for square and rectangular blocks, as shown in Figure 2. ![Wedge_Codebook_2020_01_11](./img/Wedge_Codebook_2020_01_11.png) ##### Figure 2. Wedge Codebook. Using two predictions Prediction\_1 and Prediction\_2 for the block, a final prediction p(i,j) for sample (i,j) in the block is generated by weighting the two predictions: p(i, j) = m(i, j) p1(i, j) + ( 1 - m(i, j)) p2(i, j) where m(i,j) is a function of the distance of the pixel to the wedge line. The two predictors could be both inter or one inter and one intra where intra modes are constrained to be either DC, V, H or Smooth. **Difference-based Compound Prediction** The difference-based compound prediction mode addresses cases where wedge prediction is not good enough due to for example, non-straight moving edge in a block. It considers two different predictions Prediction\_1 and Prediction\_2 of the same block, computes the pixel-wise differences between the two predictions, generates masks for each of the two predictions based on the computed pixel-wise differences, and applies the mask to the two predictions to generate the final prediction for the block. The mask for sample (i, j) is given by m(i,j) = b + a \* |p1(i, j) - p2(i, j)|, where b controls the strength of the weight, a is used to smooth the variation of the mask values around b. The prediction is generated using p(i, j) = m(i, j) \* p1(i, j) + ( 1 - m(i, j)) \* p2(i, j). Both the mask m(i,j) and (1-m(i,j)) are evaluated and the one that provides the best RD cost is selected. **Distance-based Compound Prediction** In distance-based prediction mode, the weighting applied to the two inter predictions is a function of the distance between the reference frames and the current frame. The idea is to provide more weight to the prediction from the closer reference frame. Let d0 and d1 denote the distances from the current frame to the forward and backward reference pictures. The weights depend on the ratio d1/d0, and on a set of the thresholds for the ratio. Let fwd\_offset and bck\_offset be the weights used in the distance-based compound prediction. Then we have the following: - Case where d0\>d1: The fwd\_offset and bck\_offset weights correspond to the largest Threshold value for which d1/d0\>Threshold is true. - Case where d0\<=d1: The fwd\_offset and bck\_offset weights correspond to the smallest Threshold value for which d1/d0\<Threshold is true. - When d0=0 or d1=0, if d0\<=d1 then fwd\_offset = 13 and bck\_offset = 3, else fwd\_offset = 3 and bck\_offset = 13 Table 1 below provides the weights as a function of d1/d0. ##### Table 1. fwd\_offset and back\_offset as a function of the ration d1/d0 for the case where d0\>0 and d1\>0. ![comp_mode_pred_table1](./img/comp_mode_pred_table1.png) ## 2. Implementation of the algorithm **Control macros/flags**: ##### Table 2. Inter-intra-related control flags. ![comp_mode_pred_table2](./img/comp_mode_pred_table2.png) ##### Table 3. Control flag for wedge prediction. ![comp_mode_pred_table3](./img/comp_mode_pred_table3.png) ##### Table 4. Control flags related to inter-inter compound mode prediction. ![comp_mode_pred_table4](./img/comp_mode_pred_table4.png) ### Details of the implementation The main function calls associated with compound mode prediction in mode decision are indicated in Figure 3. ![comp_mode_pred_fig5](./img/comp_mode_pred_fig5.png) ##### Figure 3. Function calls associated with compound mode prediction in mode decision. The generation of coded blocks using the compound mode involves three main steps, namely the injection of the compound mode candidates, the processing of those candidates in MD stages 0 to 3, and the final encoding of selected compound mode candidates in the encode pass. **Step 1: Injection of compound mode candidates.** The three main functions associated with compound mode prediction at the candidate injection stage are ```precompute_intra_pred_for_inter_intra```, ```inter_intra_search``` and ```determine_compound_mode```. The first two are related to the generation of inter-intra compound candidates. The third is related to the injection of inter-inter compound candidates. 1. ```Precompute_intra_pred_for_inter_intra``` The function generates for a given block DC, Vertical, Horizontal and Smooth intra predictions that would be used in subsequent stages in the compound mode candidate injection process. 2. ```Inter_intra_search``` For a given block, the generation of inter-intra wedge prediction and the smooth inter-intra prediction is performed using the function ```inter_intra_search```. The function is invoked only for the case of single reference inter predictions. The steps involved in the inter-intra search are outlined below. - Perform inter prediction through the function call ```av1_inter_prediction```. Only luma prediction is computed. - Determine if wedge prediction could be used for the given block size using the function ```is_interintra_wedge_used```. Only 8x8, 8x16, 16x8, 16x16, 16x32, 32x16, 32x32, 8x32 and 32x8 block sizes are allowed. - Enable the flag ```enable_smooth_interintra```. - Loop over the intra prediction modes: II\_DC\_PRED, II\_V\_PRED, II\_H\_PRED, II\_SMOOTH\_PRED - Perform smooth filtering of the inter prediction and the intra prediction through the function call combine\_interintra\_highbd or combine\_interintra based on the already computed inter predictions and intra predictions. The intra predictions are already generated in the function ```precompute_intra_pred_for_inter_intra```. - Compute the associated RD cost and keep track of the best RD cost and the corresponding intra prediction mode. - Perform inter-intra wedge prediction based on the best intra prediction mode from the smooth intra search step above using the function ```pick_interintra_wedge```. The details of the function are included below. ```pick_interintra_wedge```: Determines the best wedge option in the inter-intra wedge prediction. Returns the wedge index and its associated cost. - The search is allowed only for blocks sizes 8x8, 8x16, 16x8, 16x16, 16x32, 32x16, 32x32, 8x32 and 32x8. (```is_interintra_wedge_used```) - Compute the residual for intra prediction and the difference between the inter prediction and the intra prediction. (```svt_aom_highbd_subtract_block / aom_subtract_block```) - Determine the best wedge option to use based on the above computed residuals and difference. (```pick_wedge_fixed_sign```). The details of the function are included below. ```pick_wedge_fixed_sign```: Determines the best wedge option for a fixed wedge sign (0). - Check if ```inter_intra``` wedge is allowed, as described above. (```is_interintra_wedge_used```) - Loop over the available edge prediction options - Determine the mask associated with the current wedge option. (```av1_get_contiguous_soft_mask```) - Compute the corresponding prediction residuals based on the intra prediction residual and the difference between the inter prediction residuals and the intra prediction residuals. (```av1_wedge_sse_from_residuals```) - Compute the R-D cost and keep track of the best option. (```pick_wedge_fixed_sign``` and other computations.) 3. ```Determine_compound_mode``` The main function calls starting at ```Determine_compound_mode``` are outlined in Figure 4. ![comp_mode_pred_fig6](./img/comp_mode_pred_fig6.png) ##### Figure 4. Continuation of Figure 3 showing the main function calls starting with determine\_compound\_mode. The generation of ```COMPOUND_WEDGE``` and ```COMPOUND_DIFFWTD``` predictions is performed using the function ```Determine_compound\_mode```, which calls the function ```search_compound_diff_wedge```. The rest of the details are outlined in the following. For a given block, the generation of the single reference inter predictions is performed in the function ```av1_inter_prediction / av1_inter_prediction_hbd```. Only luma predictions are generated. Generate the residuals associated with the prediction from List1 reference picture, as well as the difference between the residuals corresponding to the predictions from List0 and List1 reference pictures, respectively. In the function ```pick_interinter_mask```, in the case of ```COMPOUND_WEDGE```, the function ```pick_interinter_wedge``` is called. In the case of ```COMPOUND_DIFFWTD```, the function ```pick_interinter_seg``` is called. ```pick_interinter_wedge``` generates the prediction for the case of inter-inter ```COMPOUND_WEDGE``` and updates the best ```COMPOUND_WEDGE``` prediction mode and corresponding cost. This is allowed only for block sizes 8x8, 8x16, 16x8, 16x16, 16x32, 32x16, 32x32, 8x32 and 32x8. In this function, both the nominal mask and its inverse are evaluated and the best mask is selected. The best mask also indicated the mask sign. ```pick_interinter_seg``` generates the prediction for the case of inter-inter ```COMPOUND_DIFFWTD``` and updates the best ```COMPOUND_DIFFWTD``` mask. Block size should be at least 8x8 for bipred to be allowed. As an example, consider the flow below for the function ```inject_mvp_candidates_II``` 1. Check if compound reference mode is allowed, i.e. The candidate should not be a single-reference candidate and the block size should be at least 8x8 for bipred to be allowed. 2. Determine the number of compound modes to try: - If 8x8 \<= block size \<= 32x32, then compound modes to try = compound\_types\_to\_try - else - If (compound\_types\_to\_try == MD\_COMP\_WEDGE) compound modes to try = MD\_COMP\_DIFF0 - else compound modes to try = compound\_types\_to\_try 3. Optimize further the number of modes to evaluate based on the variance of the source block. If the variance of the source block is smaller than a given threshold (```inter_inter_wedge_variance_th```), then ```MD_COMP_WEDGE``` is not considered in the search and compound modes to try is limited to MIN(compound modes to try, ```MD_COMP_DIFF0```) 4. Single reference case - Check if inter-intra is allowed: ```svt_is_interintra_allowed``` - enable\_inter\_intra flag should be set. - Block size should at least 8x8 and at most 32x32.(```is_interintra_allowed_bsize```) - Only NEARESTMV, NEARMV, GLOBALMV and NEWMV modes are allowed. (```is_interintra_allowed_mode```) - (rf\[0\] \> INTRA\_FRAME) && (rf\[1\] \<= INTRA\_FRAME). (```is_interintra_allowed_ref```); If inter\_intra is allowed, the total number of candidates to check is 3 (Single-reference inter mode, inter-intra wedge, smooth\_inter-intra), else it set to 1 (only Single-reference inter mode). - Loop over the NEARESTMV candidate and all the NEARMV candidates. - Update the candidate parameters. - Determine the intra prediction mode that yields the best smooth inter-intra prediction, and determine the best inter-intra wedge prediction option based on the best intra prediction mode from the smooth inter-intra prediction search. (```inter_intra_search```) 5. Compound reference case For all ```NEARESTMV_NEARESTMV``` and ```NEAR_NEARMV``` candidates, loop over all selected compound prediction modes - Update the candidate parameters - Determine the best wedge option for the case of ```COMPOUND_WEDGE``` or the best difference weighted prediction mask for the case of ```COMPOUND_DIFFWTD```. (```pick_interinter_mask```) **Step 2: Generate compound mode candidates in MD stages 0, 1 and 2.** The two main functions involved in generating compound mode candidates in MD stages 0, 1 and 2 are warped\_motion\_prediction and av1\_inter\_prediction. ![comp_mode_pred_fig7](./img/comp_mode_pred_fig7.png) ##### Figure 5. Continuation of Figure 3 showing the main function calls associated with compound modes in the case of warped motion prediction. 1. warped\_motion\_prediction - **plane\_warped\_motion\_prediction**: Generates the luma and chroma warped luma predictions. The chroma predictions are generated for blocks that are 16x16 or larger. - **av1\_dist\_wtd\_comp\_weight\_assign**: Returns forward offset and backward offset for the case of compound reference candidates and where the inter-inter compound prediction mde is COMPOUND\_DISTWTD. The forward offset and backward offset are used as weights in the generation of the final prediction. - **av1\_make\_masked\_warp\_inter\_predictor**: Called only in the case of compound reference candidate where the inter-inter compound type is COMPOUND\_WEDGE or COMPOUND\_DIFFWTD. Generates the predictions for both of those two compound types. The first step is to build the mask for the case of the COMPOUND\_DIFFWTD inter-inter compound type using the function av1\_build\_compound\_diffwtd\_mask\_d16. The next step is to generate the predictions using the function build\_masked\_compound\_no\_round as follows: - The function av1\_get\_compound\_type\_mask is called and returns the mask for either the case of COMPOUND\_DIFFWTD or for the case of COMPOUND\_WEDGE. The function av1\_get\_contiguous\_soft\_mask returns the mask for the case of COMPOUND\_WEDGE. For the case of COMPOUND\_DIFFWTD, the mask is computed in the step above. - The function aom\_highbd\_blend\_a64\_d16\_mask/ aom\_lowbd\_blend\_a64\_d16\_mask is the called to perform the blending of the two inter predictions using the generated mask. - **eb\_av1\_warp\_plane** is invoked in the case of BIPRED where inter-inter compound type is COMPOUND\_DISTWTD. In this case the function highbd\_warp\_plane / warp\_plane is called and in turn calls the function eb\_av1\_highbd\_warp\_affine / eb\_av1\_warp\_affine. The latter applies the affine transform and generates the warped motion prediction using the forward offset and backward offset weights associated with the COMPOUND\_DISTWTD mode. - **chroma\_plane\_warped\_motion\_prediction\_sub8x8**: Generates chroma warped motion predictions for blocks that are smaller than 16x16. The function av1\_dist\_wtd\_comp\_weight\_assign is first called to generate the mask for the COMPOUND\_DISTWTD case. The appropriate function in the function array convolve\[\]\[\]\[\] / convolveHbd\[\]\[\]\[\] is then called to generate the prediction using the forward offset and the backward offset weights. 2. **av1\_inter\_prediction** ![comp_mode_pred_fig8](./img/comp_mode_pred_fig8.png) ##### Figure 6. Continuation of Figure 3 showing the main function calls in av1_inter_prediction associated with the compound mode. In the case where inter prediction motion mode is different from ```WARPED_CAUSAL```, then the function ```av1_inter_prediction``` is called to generate the inter prediction. The main function calls associated with compound mode prediction are ```av1_dist_wtd_comp_weight_assign```, ```av1_make_masked_inter_predictor``` and ```combine_interintra```, which are described above. **Step 3**: Generate the final compound mode predictions in the encode pass. The two main relevant functions are ```warped_motion_prediction``` and ```av1_inter_prediction```. The two functions are described above. ## 3. Optimization of the algorithm **Inter-intra prediction** The settings for the different flags associated with inter-intra prediction mode are outlined in Table 5 below. ##### Table 5. Optimization settings for inter-intra compound prediction. ![comp_mode_pred_fig9](./img/comp_mode_pred_fig9.png) The flag ```md_enable_inter_intra``` is used to control when the inter-intra modes are allowed as a function of the PD pass and of the flag ```enable_inter_intra```. The latter is active in the default mode only for MR mode and for the M0 preset, otherwise it would active only if the config flag ```inter_intra_compound``` is active. **Inter-inter compound prediction** The flags ```compound_level``` and ```compound_mode``` control the complexity-quality tradeoff of the inter-inter compound prediction modes. ##### Table 6. Settings for compound\_level in inter-inter compound prediction. ![comp_mode_pred_fig10](./img/comp_mode_pred_fig10.png) ##### Table 7. Optimization settings for the inter-inter compound prediction. ![comp_mode_pred_fig11](./img/comp_mode_pred_fig11.png) The flag ```compound_types_to_try``` indicates the inter-inter compound mode to evaluate as a function of the PD pass and of the flag ```picture_control_set_ptr->parent_pcs_ptr->compound_mode```. The setting for the latter in the default mode depends on ```sequence_control_set_ptr->compound_mode```, the encoder preset and the ```sc_content_detected``` flag; otherwise, it is set to the config input value compound\_level. The flag ```sequence_control_set_ptr->compound_mode``` depends in the default configuration on the encoder preset, otherwise it set to the config input value ```compound_level```. **Inter-inter wedge prediction** For the case of inter-inter wedge prediction, the flag ```wedge_mode``` decides on the tradeoff between complexity and quality for inter-inter wedge prediction. The settings for the flag are given in Table 8. ##### Table 8. wedge\_mode settings and description. ![comp_mode_pred_fig12](./img/comp_mode_pred_fig12.png) Currently, wedge\_mode is set to 0, i.e. full search is performed all the time as indicated in Table 9. ##### Table 9. wedge\_mode settings. ![comp_mode_pred_fig13](./img/comp_mode_pred_fig13.png) Whether to include wedge prediction in the case of inter-inter compound prediction is also controlled by the variance of the source block. If the variance of the source block is smaller than a given threshold (```inter_inter_wedge_variance_th```), then ```MD_COMP_WEDGE``` is not considered in the search and the compound modes to try are limited to at most to ```MD_COMP_DIST``` and ```MD_COMP_DIFF0```. ## Notes The feature settings that are described in this document were compiled at v0.8.3 of the code and may not reflect the current status of the code. The description in this document represents an example showing how features would interact with the SVT architecture. For the most up-to-date settings, it's recommended to review the section of the code implementing this feature. ## References \[1\] Cheng Chen, Jingning Han, and Yaowu Xu, “A Hybrid Weighted Compound Motion Compensated Prediction for Video Compression,” Picture Coding Symposium, pp. 223-227, 2018. \[2\] Yue Chen, Debargha Murherjee, Jingning Han, Adrian Grange, Yaowu Xu, Zoe Liu, Sarah Parker, Cheng Chen, Hui Su, Urvang Joshi, Ching-Han Chiang, Yunqing Wang, Paul Wilkins, Jim Bankoski, Luc Trudeau, Nathan Egge, Jean-Marc Valin, Thomas Davies, Steinar Midtskogen, Andrey Norkin and Peter de Rivaz, “An Overview of Core Coding Tools in the AV1 Video Codec,” Picture Coding Symposium, pp. 41-45, 2018.