From bb432b21f9ae8a84d29c8940e92c7a5b6186850e Mon Sep 17 00:00:00 2001
From: Colombo <lepersorium@gmail.com>
Date: Wed, 25 Nov 2020 12:19:43 +0400
Subject: [PATCH] .

---
 core/imagelib/warp.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/core/imagelib/warp.py b/core/imagelib/warp.py
index 9252532..ac4f324 100644
--- a/core/imagelib/warp.py
+++ b/core/imagelib/warp.py
@@ -41,6 +41,9 @@ def gen_warp_params (w, flip, rotation_range=[-10,10], scale_range=[-0.5, 0.5],
     params['mapx'] = mapx
     params['mapy'] = mapy
     params['rmat'] = random_transform_mat
+    u_mat = random_transform_mat.copy()
+    u_mat[:,2] /= w
+    params['umat'] = u_mat
     params['w'] = w
     params['rw'] = rw
     params['flip'] = p_flip