crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix var name

Browse Source
This commit is contained in:
Junhao Liang
2023-09-27 17:35:20 +08:00
committed by GitHub
parent 2c29afadf3
commit 36d36ed471
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
opengait/data/transform.py
+3 -3
View File
@@ -236,10 +236,10 @@ class MirrorPoses(object):
Performing Mirror Operations Performing Mirror Operations
""" """
def __init__(self, prob=0.5): def __init__(self, prob=0.5):
self.probability = probability self.prob = prob
def __call__(self, data): def __call__(self, data):
if np.random.random() <= self.probability: if np.random.random() <= self.prob:
center = np.mean(data[:, :, 0], axis=1, keepdims=True) center = np.mean(data[:, :, 0], axis=1, keepdims=True)
data[:, :, 0] = center - data[:, :, 0] + center data[:, :, 0] = center - data[:, :, 0] + center
@@ -449,4 +449,4 @@ class MSGGTransform():
def __call__(self, x): def __call__(self, x):
result=x[...,self.mask,:].copy() result=x[...,self.mask,:].copy()
return result return result