vhad/SoftGroup
1
0
Fork 0
mirror of https://github.com/botastic/SoftGroup.git synced 2025-10-16 11:45:42 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

minor

Browse Source
This commit is contained in:
Thang Vu 2022-04-15 05:37:01 +00:00
parent efe2ddb6b1
commit 3b39cc3bf1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
softgroup/data/custom.py
View File

@ -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]])