Ejemplo n.º 1
0
 def backward_pixel_map(ctx, faces, face_index_map, rgb_map, alpha_map,
                        grad_rgb_map, grad_alpha_map, grad_faces):
     if (not ctx.return_rgb) and (not ctx.return_alpha):
         return grad_faces
     else:
         return rasterize_cuda.backward_pixel_map(
             faces, face_index_map, rgb_map, alpha_map, grad_rgb_map,
             grad_alpha_map, grad_faces, ctx.image_size, ctx.eps,
             ctx.return_rgb, ctx.return_alpha)
Ejemplo n.º 2
0
 def backward_pixel_map(self, faces, face_index_map, rgb_map, alpha_map,
                        grad_rgb_map, grad_alpha_map, grad_faces):
     if (not self.return_rgb) and (not self.return_alpha):
         return grad_faces
     else:
         return rasterize_cuda.backward_pixel_map(
             faces, face_index_map, rgb_map, alpha_map, grad_rgb_map,
             grad_alpha_map, grad_faces, self.image_size, self.eps,
             int(self.return_rgb), int(self.return_alpha))