Spaces:
Sleeping
Sleeping
Update difpoint/inference.py
Browse files- 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 |
}
|