Advertisement
Guest User

Untitled

a guest
Apr 6th, 2020
233
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.65 KB | None | 0 0
  1. def name():
  2.     return ''.join([chr(e[0]) for e in f[f['anno'][0][0]].value])
  3.  
  4. if __name__ == '__main__':
  5.     path = 'anno.mat'
  6.     import h5py
  7.  
  8.     with h5py.File(path, 'r') as f:
  9.         for i in range(f['anno'].shape[1]):
  10.             name_ref = f['anno'][0][i]
  11.             name_container = f[name_ref]
  12.             name = ''.join([chr(e[0]) for e in name_container.value])
  13.  
  14.             bbox = ((f[f[f['anno'][1][i]].value[0][0]].value.ravel()) + 0.5).astype(int).tolist()
  15.  
  16.             angles = f[f[f['anno'][2][i]].value[0][0]].value.ravel().tolist()
  17.  
  18.             print(name)
  19.             print(bbox)
  20.             print(angles)
  21.             break
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement