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, "learning_rate": 0.01,
"min_learning_rate": 1e-05, "min_learning_rate": 1e-05,
"learning_decay_rate": 0.9999, "learning_decay_rate": 0.9999,
"prediction_horizon": 50, "prediction_horizon": 30,
"minimum_history_length": 10, "minimum_history_length": 5,
"maximum_history_length": 80, "maximum_history_length": 50,
"map_encoder": { "map_encoder": {
"PEDESTRIAN": { "PEDESTRIAN": {
"heading_state_index": [2, 3], "heading_state_index": [2, 3],
@ -74,7 +74,7 @@
"dynamic": { "dynamic": {
"PEDESTRIAN": { "PEDESTRIAN": {
"name": "SingleIntegrator", "name": "SingleIntegrator",
"distribution": false, "distribution": true,
"limits": {} "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) print(max_frame_nr)
# separate call so cursor is kept during multiple loops # separate call so cursor is kept during multiple loops
seed(123) # seed(123)
shuffle(tracks) shuffle(tracks)
dt1 = RollingAverage() dt1 = RollingAverage()