Required argument net is missing when calling function shade_images.

nv-tlabs · Nov 9, 2021 · beeea55 · beeea55
1 parent 6cb8d10
commit beeea55
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/sdf-net/app/sdf_renderer.py b/sdf-net/app/sdf_renderer.py
@@ -138,7 +138,8 @@ def write_exr(path, data):
             rad = np.radians(angle)
             model_matrix = torch.FloatTensor(R.from_rotvec(rad * np.array([0, 1, 0])).as_matrix())
 
-            out = renderer.shade_images(f=args.camera_origin,
+            out = renderer.shade_images(net=net,
+                                        f=args.camera_origin,
                                         t=args.camera_lookat,
                                         fov=args.camera_fov,
                                         aa=not args.disable_aa,
@@ -159,7 +160,8 @@ def write_exr(path, data):
         views = sample_fib_sphere(args.nb_poses)
         cam_origins = args.cam_radius * views
         for p, cam_origin in enumerate(cam_origins):
-            out = renderer.shade_images(f=cam_origin,
+            out = renderer.shade_images(net=net,
+                                        f=cam_origin,
                                         t=args.camera_lookat,
                                         fov=args.camera_fov,
                                         aa=not args.disable_aa,