Add Ground Truth Estimator

Change-Id: Iec6c7e49a64610e33a77c7d5d772e6b063a0f1e0

1 files changed, 36 insertions, 0 deletions

diff --git a/tools/3D-Reconstruction/MotionEST/GroundTruth.py b/tools/3D-Reconstruction/MotionEST/GroundTruth.py
new file mode 100644
index 000000000..61b4bef42
--- /dev/null
+++ b/tools/3D-Reconstruction/MotionEST/GroundTruth.py
@@ -0,0 +1,36 @@
+#!/usr/bin/env python
+# coding: utf-8
+import numpy as np
+import numpy.linalg as LA
+from MotionEST import MotionEST
+"""Ground Truth:
+
+    Load in ground truth motion field and mask
+"""
+
+
+class GroundTruth(MotionEST):
+  """
+    constructor:
+        cur_f: current frame
+        ref_f: reference frame
+        blk_sz: block size
+        gt_path: ground truth motion field file path
+    """
+
+  def __init__(self, cur_f, ref_f, blk_sz, gt_path):
+    self.name = 'ground truth'
+    super(GroundTruth, self).__init__(cur_f, ref_f, blk_sz)
+    self.mask = np.zeros((self.num_row, self.num_col), dtype=np.bool)
+    with open(gt_path) as gt_file:
+      lines = gt_file.readlines()
+      for i in xrange(len(lines)):
+        info = lines[i].split(';')
+        for j in xrange(len(info)):
+          x, y = info[j].split(',')
+          #-,- stands for nothing
+          if x == '-' or y == '-':
+            self.mask[i, -j - 1] = True
+            continue
+          #the order of original file is flipped on the x axis
+          self.mf[i, -j - 1] = np.array([float(y), -float(x)], dtype=np.int)