From: Angie Chiang <angiebird@google.com>
Date: Thu, 4 Apr 2019 21:44:09 +0000 (-0700)
Subject: Print mv_mode counts
X-Git-Tag: v1.8.1~124^2
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=62d33725ec0f2a46baa2e6a6c331de36b5e11033;p=libvpx

Print mv_mode counts

Change-Id: I9d315e359e384dc0295c3471d8179bd828fddc1b
---

diff --git a/tools/non_greedy_mv/non_greedy_mv.py b/tools/non_greedy_mv/non_greedy_mv.py
index 5ec5f4ad4..513faa435 100644
--- a/tools/non_greedy_mv/non_greedy_mv.py
+++ b/tools/non_greedy_mv/non_greedy_mv.py
@@ -177,4 +177,10 @@ if __name__ == '__main__':
     #axes[1][1].figure.colorbar(im, ax=axes[1][1])
 
     print rf_idx, frame_idx, ref_frame_idx, gf_frame_offset, ref_gf_frame_offset, len(mv_ls)
+
+    flatten_mv_mode = mv_mode_arr.flatten()
+    zero_mv_count = sum(flatten_mv_mode == 0);
+    new_mv_count = sum(flatten_mv_mode == 1);
+    ref_mv_count = sum(flatten_mv_mode == 2) + sum(flatten_mv_mode == 3);
+    print zero_mv_count, new_mv_count, ref_mv_count
     plt.show()