update

Author: Fangyh09

estimator.py

@@ -32,8 +32,9 @@ def __init__(self, model_path, net, inp_dim=224):
 
     def addlayer(self, image):
         h, w = image.shape[:2]
-        x = np.arange(0, w)
-        y = np.arange(0, h) 
+        # todo retraining model
+        x = np.arange(0, h)
+        y = np.arange(0, w) 
         x, y = np.meshgrid(x, y)
         x = x[:,:, np.newaxis]
         y = y[:,:, np.newaxis]