Daniellesry commited on
Commit
45ed92a
·
1 Parent(s): 2f07cca
Files changed (1) hide show
  1. dkt/pipelines/pipeline.py +1 -1
dkt/pipelines/pipeline.py CHANGED
@@ -1019,7 +1019,7 @@ class DKTPipeline:
1019
 
1020
  print(f'moge devices: {moge_device}') #* why cpu?
1021
 
1022
- output = self.moge_pipe.infer(input_image)
1023
 
1024
  #* "dict_keys(['points', 'intrinsics', 'depth', 'mask', 'normal'])"
1025
  moge_intrinsics = output['intrinsics'].cpu().numpy()
 
1019
 
1020
  print(f'moge devices: {moge_device}') #* why cpu?
1021
 
1022
+ output = self.moge_infer(input_image)
1023
 
1024
  #* "dict_keys(['points', 'intrinsics', 'depth', 'mask', 'normal'])"
1025
  moge_intrinsics = output['intrinsics'].cpu().numpy()