Update mono/utils/do_test.py

mono/utils/do_test.py

@@ -211,12 +211,12 @@ def transform_test_data_scalecano(rgb, intrinsic, data_basic):
 
  rgb = torch.from_numpy(rgb.transpose((2, 0, 1))).float()
  rgb = torch.div((rgb - mean), std)
- #rgb = rgb[None, :, :, :].cuda()
- rgb = rgb[None, :, :, :]
+ rgb = rgb[None, :, :, :].cuda()
+ #rgb = rgb[None, :, :, :]
  
  cam_model = torch.from_numpy(cam_model.transpose((2, 0, 1))).float()
- #cam_model = cam_model[None, :, :, :].cuda()
- cam_model = cam_model[None, :, :, :]
+ cam_model = cam_model[None, :, :, :].cuda()
+ #cam_model = cam_model[None, :, :, :]
  cam_model_stacks = [
  torch.nn.functional.interpolate(cam_model, size=(cam_model.shape[2]//i, cam_model.shape[3]//i), mode='bilinear', align_corners=False)
  for i in [2, 4, 8, 16, 32]