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

update angles output

Browse Source
This commit is contained in:
Lorenzo 2020-09-29 20:45:22 +02:00
parent 2cbc4a23c1
commit fac9ff1d86
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
monstereo/network/net.py
View File

@ -212,10 +212,15 @@ class Loco:
dic_out['uv_shoulders'].append(uv_shoulder) dic_out['uv_shoulders'].append(uv_shoulder)
dic_out['uv_heads'].append(uv_head) dic_out['uv_heads'].append(uv_head)
# For MonStereo / MonoLoco++
try:
dic_out['angles'].append(float(dic_in['yaw'][0][idx])) # Predicted angle
dic_out['angles_egocentric'].append(float(dic_in['yaw'][1][idx])) # Egocentric angle
except KeyError:
continue
# Only for MonStereo # Only for MonStereo
try: try:
angle = float(dic_in['yaw'][0][idx]) # Predicted angle
dic_out['angles'].append(angle)
dic_out['aux'].append(float(dic_in['aux'][idx])) dic_out['aux'].append(float(dic_in['aux'][idx]))
except KeyError: except KeyError:
continue continue