From 3b39cc3bf1d8f9e427ccb3d25c38e80d217bb123 Mon Sep 17 00:00:00 2001 From: Thang Vu Date: Fri, 15 Apr 2022 05:37:01 +0000 Subject: [PATCH] minor --- softgroup/data/custom.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/softgroup/data/custom.py b/softgroup/data/custom.py index 2c15e9f..3818ebd 100644 --- a/softgroup/data/custom.py +++ b/softgroup/data/custom.py @@ -97,7 +97,7 @@ class CustomDataset(Dataset): [-math.sin(theta), math.cos(theta), 0], [0, 0, 1]]) else: # Empirically, slightly rotate the scene can match the results from checkpoint - theta = 0.45 * math.pi + theta = 0.35 * math.pi m = np.matmul(m, [[math.cos(theta), math.sin(theta), 0], [-math.sin(theta), math.cos(theta), 0], [0, 0, 1]])