Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix augmentation bug

Browse Source
This commit is contained in:
Yin Li 2020-01-21 18:49:45 -05:00
parent 1f89e894cc
commit 46a3a3a97d
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
map2map/data/fields.py
View File

@ -156,8 +156,8 @@ def flip(fields, axes, ndim):
if x.shape[0] == ndim: # flip vector components if x.shape[0] == ndim: # flip vector components
x[axes] = - x[axes] x[axes] = - x[axes]
axes = (1 + axes).tolist() shifted_axes = (1 + axes).tolist()
x = torch.flip(x, axes) x = torch.flip(x, shifted_axes)
new_fields.append(x) new_fields.append(x)
@ -172,8 +172,8 @@ def perm(fields, axes, ndim):
if x.shape[0] == ndim: # permutate vector components if x.shape[0] == ndim: # permutate vector components
x = x[axes] x = x[axes]
axes = [0] + (1 + axes).tolist() shifted_axes = [0] + (1 + axes).tolist()
x = x.permute(axes) x = x.permute(shifted_axes)
new_fields.append(x) new_fields.append(x)