Skip to content

Latest commit

 

History

History
24 lines (14 loc) · 479 Bytes

README.md

File metadata and controls

24 lines (14 loc) · 479 Bytes

ETA : Encourage Time for Attention

Description

ETA project의 집중 상태 판단 모델. ViT 기반으로 학습을 진행하였습니다.

Our Model Output

result

Install

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

How to Inference

Please Check inference.ipynb