Generative Query Network (GQN) in PyTorch as described in "Neural Scene Representation and Rendering"
Provides pretrained model weights for a GQN trained on the Shepard-Metzler 5-parts dataset as provided in the mental-rotation.ipynb notebook.
mental-rotation.ipynb