speed aug fixed

This commit is contained in:
Si11ium 2020-05-21 14:44:05 +02:00
parent 8cec323286
commit 4b4051c045

View File

@ -10,9 +10,9 @@ import numpy as np
class Speed(object): class Speed(object):
def __init__(self, max_amount=0.3, speed_min=1, speed_max=1): def __init__(self, max_amount=0.3, speed_min=1, speed_max=1):
self.speed_max = speed_max self.speed_max = speed_max if speed_max else 1
self.speed_min = speed_min if speed_min else 1 self.speed_min = speed_min if speed_min else 1
self.max_amount = speed_max if speed_max else 1 self.max_amount = min(max(0, max_amount), 1)
def __call__(self, x): def __call__(self, x):
if self.speed_min == 1 and self.speed_max == 1: if self.speed_min == 1 and self.speed_max == 1: