mirror of
https://git.intern.spaceteamaachen.de/ALPAKA/SPATZ.git
synced 2025-12-17 01:38:02 +00:00
Updated observers and Kalman Filter, added running average
This commit is contained in:
@@ -11,10 +11,12 @@ from spatz.dataset import *
|
||||
|
||||
|
||||
class Sensor:
|
||||
def __init__(self, dataset: Dataset, logger: Logger, transforms: List[Transform] = []):
|
||||
def __init__(self, dataset: Dataset, logger: Logger, transforms: List[Transform] = [], min_value=-np.inf, max_value=np.inf):
|
||||
self._dataset = dataset
|
||||
self._logger = logger
|
||||
self._transforms = transforms
|
||||
self._min_value = min_value
|
||||
self._max_value = max_value
|
||||
|
||||
def set_dataset(self, dataset: Dataset):
|
||||
self._dataset = dataset
|
||||
@@ -52,6 +54,8 @@ class Sensor:
|
||||
for transform in self._transforms:
|
||||
out = transform(t, out)
|
||||
|
||||
out = np.clip(out, self._min_value, self._max_value)
|
||||
|
||||
# Log the outputs of the sensor.
|
||||
if np.isscalar(out):
|
||||
self._log('out', out)
|
||||
|
||||
Reference in New Issue
Block a user