Skip to content

KimTaeYun02/ETA

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

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

About

2023 Capstone

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published