A simple Python raytracer that supports spheres with configurable "material" properties (base color and a bunch of
light coefficients). To generate a raytraced image of the pre-defined scene, run: python raytracer.py and open
image.ppm with a PPM-compatible viewer (eog works fine on Linux):
I found the following resources extremely helpful:
| import torch.utils.data as data | |
| import os | |
| import re | |
| import torch | |
| import tarfile | |
| from PIL import Image | |
| IMG_EXTENSIONS = ['.png', '.jpg', '.jpeg'] |
| def Rop(y, x, v): | |
| """Computes an Rop. | |
| Arguments: | |
| y (Variable): output of differentiated function | |
| x (Variable): differentiated input | |
| v (Variable): vector to be multiplied with Jacobian from the right | |
| """ | |
| w = torch.ones_like(y, requires_grad=True) | |
| return torch.autograd.grad(torch.autograd.grad(y, x, w), w, v) |
