ETA project의 집중 상태 판단 모델. ViT 기반으로 학습을 진행하였습니다.
Before running this program, please your pytorch version.
This code was tested with python 3.8.x
, pytoch 1.14
$ git clone https://github.com/KimTaeYun02/ETA.git
$ pip install pytorch_pretrained_vit
Please Check inference.ipynb