shithub: libvpx

Download patch

ref: 78c44e2dc26e383fdc54eb4bf406a76e52ea361e
parent: b4d593381b6d8008bedf90359585369a847b767c
author: Jerome Jiang <jianj@google.com>
date: Mon May 13 07:29:55 EDT 2019

Reland "vp9: Enable ml based partition for speed>=8 low res."

Disable in high bitdepth build.

This reverts commit 152358da777d45613b386be3a116719b8b16bd6d.

Change-Id: I9996d0963915ed4db0fde80c6290d91b3ce63719

--- a/vp9/encoder/vp9_speed_features.c
+++ b/vp9/encoder/vp9_speed_features.c
@@ -569,14 +569,6 @@
         (frames_since_key % (sf->last_partitioning_redo_frequency << 1) == 1);
     sf->max_delta_qindex = is_keyframe ? 20 : 15;
     sf->partition_search_type = REFERENCE_PARTITION;
-    if (sf->nonrd_use_ml_partition) {
-      if (!frame_is_intra_only(cm) && cm->width >= 360 && cm->height >= 360)
-        sf->partition_search_type = ML_BASED_PARTITION;
-#if CONFIG_VP9_HIGHBITDEPTH
-      if (cpi->Source->flags & YV12_FLAG_HIGHBITDEPTH)
-        sf->partition_search_type = REFERENCE_PARTITION;
-#endif  // CONFIG_VP9_HIGHBITDEPTH
-    }
     if (cpi->oxcf.rc_mode == VPX_VBR && cpi->oxcf.lag_in_frames > 0 &&
         cpi->rc.is_src_frame_alt_ref) {
       sf->partition_search_type = VAR_BASED_PARTITION;
@@ -641,10 +633,7 @@
       sf->use_altref_onepass = 1;
       sf->use_compound_nonrd_pickmode = 1;
     }
-
-    if (!sf->nonrd_use_ml_partition)
-      sf->partition_search_type = VAR_BASED_PARTITION;
-
+    sf->partition_search_type = VAR_BASED_PARTITION;
     sf->mv.search_method = NSTEP;
     sf->mv.reduce_first_step_size = 1;
     sf->skip_encode_sb = 0;
@@ -732,6 +721,14 @@
     if (!cpi->use_svc) cpi->max_copied_frame = 4;
     if (cpi->row_mt && cpi->oxcf.max_threads > 1)
       sf->adaptive_rd_thresh_row_mt = 1;
+    // Enable ML based partition for low res.
+    if (!frame_is_intra_only(cm) && cm->width * cm->height <= 352 * 288) {
+      sf->nonrd_use_ml_partition = 1;
+    }
+#if CONFIG_VP9_HIGHBITDEPTH
+    if (cpi->Source->flags & YV12_FLAG_HIGHBITDEPTH)
+      sf->nonrd_use_ml_partition = 0;
+#endif
     if (content == VP9E_CONTENT_SCREEN) sf->mv.subpel_force_stop = HALF_PEL;
     // Only keep INTRA_DC mode for speed 8.
     if (!is_keyframe) {
@@ -777,6 +774,9 @@
     if (cpi->rc.avg_frame_low_motion < 70) sf->default_interp_filter = BILINEAR;
     if (cm->width * cm->height >= 640 * 360) sf->variance_part_thresh_mult = 2;
   }
+
+  if (sf->nonrd_use_ml_partition)
+    sf->partition_search_type = ML_BASED_PARTITION;
 
   if (sf->use_altref_onepass) {
     if (cpi->rc.is_src_frame_alt_ref && cm->frame_type != KEY_FRAME) {