privi-gaze-distill / models /__init__.py
BcantCode's picture
Upload models/__init__.py
12b8ffa verified
raw
history blame contribute delete
687 Bytes
"""
PriviGaze: Privileged Distillation for Accessible Gaze Estimation
Models for gaze estimation using privileged knowledge distillation.
- Teacher: Siamese network with RGB eye crops + blurred face (privileged info)
- Student: Ultra-compact CNN with only grayscale face (on-device deployment)
"""
from .teacher import PriviGazeTeacher
from .student import PriviGazeStudent, count_parameters
from .distillation_loss import PriviGazeDistillationLoss
from .dataset import SyntheticGazeDataset, create_dataloaders
__all__ = [
'PriviGazeTeacher',
'PriviGazeStudent',
'count_parameters',
'PriviGazeDistillationLoss',
'SyntheticGazeDataset',
'create_dataloaders',
]