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]