summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--vp9/encoder/vp9_encodeframe.c2
1 files changed, 2 insertions, 0 deletions
diff --git a/vp9/encoder/vp9_encodeframe.c b/vp9/encoder/vp9_encodeframe.c
index 825d8d9ac..bc0691bc1 100644
--- a/vp9/encoder/vp9_encodeframe.c
+++ b/vp9/encoder/vp9_encodeframe.c
@@ -3604,9 +3604,11 @@ static int wiener_var_segment(VP9_COMP *cpi, BLOCK_SIZE bsize, int mi_row,
for (col = mb_col_start; col < mb_col_end; ++col)
wiener_variance += cpi->mb_wiener_variance[row * cm->mb_cols + col];
+ wiener_variance /= (mb_row_end - mb_row_start) * (mb_col_end - mb_col_start);
#if CONFIG_MULTITHREAD
pthread_mutex_lock(&cpi->kmeans_mutex);
#endif // CONFIG_MULTITHREAD
+
kmeans_data = &cpi->kmeans_data_arr[cpi->kmeans_data_size++];
kmeans_data->value = log(1.0 + wiener_variance) / log(2.0);
kmeans_data->pos = mi_row * cpi->kmeans_data_stride + mi_col;