search

Ahmednull / L2CS-Net

The official PyTorch implementation of L2CS-Net for gaze estimation and tracking
MIT License
343 stars 81 forks source link

Packaging code and creating high-level API #18

Closed edavalosanaya closed 1 year ago

edavalosanaya commented 1 year ago

Hello!

This is my first PR to this repo!

I made 4 commits for this PR. Let me elaborate on each and let me know what you think.

  1. (refactor to package): Instead of using requirements.txt, I added a pyproject.toml and wrapped the project's code into an L2CS package. Now you can install the code via pip install . and use it as an importable package.
  2. Added CPU mode functionality
  3. Wrapped code into Pipeline class, to provide a nice and clean API
  4. Fixed any issues and updated the README.md
  5. Fixes for multiple head detection (batched format) and visualization of gaze vector (controlling better the gaze vector length).