YinuoGuo27 commited on
Commit
33a3a16
·
verified ·
1 Parent(s): 04a2083

Update difpoint/inference.py

Browse files
Files changed (1) hide show
  1. difpoint/inference.py +1 -1
difpoint/inference.py CHANGED
@@ -376,7 +376,7 @@ class Inferencer(object):
376
 
377
  item_dct = {
378
  'scale': x_d_i_info['scale'][i].cpu().numpy().astype(np.float32),
379
- 'R_d': R_d_i.astype(np.float32),
380
  'exp': x_d_i_info['exp'][i].reshape(1, 21, -1).cpu().numpy().astype(np.float32),
381
  't': x_d_i_info['t'][i].cpu().numpy().astype(np.float32),
382
  }
 
376
 
377
  item_dct = {
378
  'scale': x_d_i_info['scale'][i].cpu().numpy().astype(np.float32),
379
+ 'R_d': R_d_i.cpu().numpy().astype(np.float32),
380
  'exp': x_d_i_info['exp'][i].reshape(1, 21, -1).cpu().numpy().astype(np.float32),
381
  't': x_d_i_info['t'][i].cpu().numpy().astype(np.float32),
382
  }