/Where-are-they-looking-PyTorch

Where are they looking? - Attention modelling + Deep Learning

Primary LanguagePythonMIT LicenseMIT

Where are they looking?

This repository contains an implementation of the "Where are they looking?" paper by A. Recasens*, A. Khosla*, C. Vondrick and A. Torralba.

Introduction

A deep neural network-based approach for gaze-following automated using a SSD face detector.

Installation

Pretrained Models

Usage

python3 main.py --data_dir=location to gazefollow dataset --placesmodelpath=location to places365 alexnet model

  • Please check out opts.py for other parameter changing.

Contact

Please do get in touch with us by email for any questions, comments, suggestions you have!

References