/foveated-convolutions

Code for the paper: Foveated Convolutions: Improving Spatial Transformer Networks by Modelling the Retina

Primary LanguagePythonMIT LicenseMIT

SVRHM • NeurIPS 2019

Foveated Convolutions: Improving Spatial Transformer Networks by Modelling the Retina

Ethan Harris (@ethanwharris), Mahesan Niranjan and Jonathon Hare (@jonhare)

About

This repository contains the code for our paper 'Foveated Convolutions: Improving Spatial Transformer Networks by Modelling the Retina', accepted to the NeurIPS 2019 workshop on Shared Visual Representations in Humans and Machines (SVRHM).

Code

The main file for running experiments is main.py, uncomment the relevant sections to run different configurations. ResNet experiments and the code to generate the figure can be found in train_resnet.py and figure.py respectively.