security_vision/trap
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Tweak to have more trajectories to train from

Browse source
This commit is contained in:
Ruben van de Ven 2024-12-31 14:47:33 +01:00
parent 71eadbeecf
commit 8db0f2cf6c
2 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
EXPERIMENTS/config.json
View file

@ -5,9 +5,9 @@
"learning_rate": 0.01,
"min_learning_rate": 1e-05,
"learning_decay_rate": 0.9999,
"prediction_horizon": 50,
"minimum_history_length": 10,
"maximum_history_length": 80,
"prediction_horizon": 30,
"minimum_history_length": 5,
"maximum_history_length": 50,
"map_encoder": {
"PEDESTRIAN": {
"heading_state_index": [2, 3],
@ -74,7 +74,7 @@
"dynamic": {
"PEDESTRIAN": {
"name": "SingleIntegrator",
"distribution": false,
"distribution": true,
"limits": {}
}
},

2
trap/process_data.py
View file

@ -145,7 +145,7 @@ def process_data(src_dir: Path, dst_dir: Path, name: str, smooth_tracks: bool, c
print(max_frame_nr)
# separate call so cursor is kept during multiple loops
seed(123)
# seed(123)
shuffle(tracks)
dt1 = RollingAverage()