fix var name

This commit is contained in:
Junhao Liang
2023-09-27 17:35:20 +08:00
committed by GitHub
parent 2c29afadf3
commit 36d36ed471
+3 -3
View File
@@ -236,10 +236,10 @@ class MirrorPoses(object):
Performing Mirror Operations
"""
def __init__(self, prob=0.5):
self.probability = probability
self.prob = prob
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)
data[:, :, 0] = center - data[:, :, 0] + center
@@ -449,4 +449,4 @@ class MSGGTransform():
def __call__(self, x):
result=x[...,self.mask,:].copy()
return result
return result