fix var name
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user