Skip to content

Commit ac9d44a

Browse files
committed
dowscale not all the time
1 parent 28753dc commit ac9d44a

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

simlearner3d/processing/transforms/augmentations.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,8 +101,9 @@ def __init__(self,scales):
101101
self.scales=scales
102102

103103
def __call__(self, data: Data):
104-
scale_factor=random.choice(self.scales)
105-
data=self.apply_scale(data,scale_factor)
104+
if random.random()< 0.5 :
105+
scale_factor=random.choice(self.scales)
106+
data=self.apply_scale(data,scale_factor)
106107
return data
107108

108109
def apply_scale(self,data: Data,scale_factor):

0 commit comments

Comments
 (0)