Update transform.py
This commit is contained in:
@@ -63,7 +63,7 @@ class RandomHorizontalFlip(object):
|
|||||||
if random.uniform(0, 1) >= self.prob:
|
if random.uniform(0, 1) >= self.prob:
|
||||||
return seq
|
return seq
|
||||||
else:
|
else:
|
||||||
return seq[:, :, ::-1]
|
return seq[..., ::-1]
|
||||||
|
|
||||||
|
|
||||||
class RandomErasing(object):
|
class RandomErasing(object):
|
||||||
|
|||||||
Reference in New Issue
Block a user