latest

pbnr_pytorch/scene_loaders/ibr_scene.py

@@ -64,7 +64,7 @@ class Camera(nn.Module):
             self.image = nn.Parameter(loaded_optimizedimages.requires_grad_(True))
         else:
             image = image.clamp(0.0, 1.0)
-            self.image = nn.Parameter(image.requires_grad_(True))
+            self.image = nn.Parameter(image.cuda().requires_grad_(True))
 
         self.image_width = self.image.shape[3]
         self.image_height = self.image.shape[2]

pbnr_pytorch/test_render_and_save.py

@@ -99,9 +99,9 @@ with torch.no_grad():
     torch.manual_seed(0)
     random.seed(0)
 
-    dataset_name = "hallway_lamp"
+    dataset_name = "silver_vase2"
     path_name = "test_path2"
-    args.input_path = "/data/graphdeco/user/gkopanas/scenes/catacaustic_new/" + dataset_name + "/sibr/pbnrScene/" + dataset_name + ".json"
+    args.input_path = "/data/graphdeco/user/gkopanas/scenes/new_selection/" + dataset_name + "/sibr/pbnrScene/" + dataset_name + ".json"
 
     with open(args.input_path) as json_file:
         input_json = json.load(json_file)
@@ -118,9 +118,9 @@ with torch.no_grad():
                            int(args.test_cameras), load_iter)
 
 
-        test_cameras_info = readColmapSceneInfo_Test(dataset_name, r"/data/graphdeco/user/gkopanas/scenes/catacaustic_new/" + dataset_name + "/colmap_1000/" + path_name).cameras
+        test_cameras_info = readColmapSceneInfo_Test(dataset_name, r"/data/graphdeco/user/gkopanas/scenes/new_selection/" + dataset_name + "/colmap_1000/" + path_name).cameras
         test_cameras = []
-        test_neighbors_dict_path = os.path.join(r"/data/graphdeco/user/gkopanas/scenes/catacaustic_new/" + dataset_name + "/sibr/pbnrScene/" + "neighbors_dict_test.json")
+        test_neighbors_dict_path = os.path.join(r"/data/graphdeco/user/gkopanas/scenes/new_selection/" + dataset_name + "/sibr/pbnrScene/" + "neighbors_dict_test.json")
         with open(test_neighbors_dict_path) as json_file:
             test_neighbors_dict = json.load(json_file)