From cd13f6fd788892989dfb2fc4bbd3d3d78f9bc3df Mon Sep 17 00:00:00 2001 From: sunhz Date: Tue, 19 Dec 2023 10:50:11 +0800 Subject: [PATCH] FIX: incorrectly calling the pyTorch method --- demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo.py b/demo.py index d1267d3..3a4795e 100644 --- a/demo.py +++ b/demo.py @@ -164,7 +164,7 @@ def demo2(): ), axis=1) q2 = p2 + np.random.randint(-20, 20, size=p2.shape) - rigid2 = mls_rigid_deformation_pt(vy, vx, p2, q2, alpha=1) + rigid2 = mls_rigid_deformation(vy, vx, p2, q2, alpha=1) aug2 = np.ones_like(image) aug2[vx, vy] = image[tuple(rigid2)]