charles/monoloco
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

CASR model passed through args with std gestures

Browse Source
This commit is contained in:
Charles Joseph Pierre Beauville 2021-06-28 10:04:31 +02:00
parent 872aec93e1
commit 8ac304a430
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
monoloco/network/net.py
View File

@ -61,6 +61,9 @@ class Loco:
if casr == 'std': if casr == 'std':
print("CASR with standard gestures") print("CASR with standard gestures")
turning_output_size = 3 turning_output_size = 3
if casr_model:
turning_model_path = casr_model
else:
turning_model_path = "/home/beauvill/Repos/monoloco/data/outputs/casr_standard-210613-0005.pkl" turning_model_path = "/home/beauvill/Repos/monoloco/data/outputs/casr_standard-210613-0005.pkl"
elif casr == 'nonstd': elif casr == 'nonstd':
turning_output_size = 4 turning_output_size = 4