tomas-gajarsky/facetorch-fer-efficientnet-b2

Facial Expression Recognition (8 classes)

Exported PyTorch model (.pt2) for use with facetorch.

Model Details

Task	Facial Expression Recognition (8 classes)
Architecture	EfficientNet-B2 (TF-style, max pooling)
Format	torch.export (.pt2) — no model source code needed
Dynamic shapes	Batch dimension is dynamic (1-64)
Input	260x260 face crop
Output	8-dim logits: Anger, Contempt, Disgust, Fear, Happiness, Neutral, Sadness, Surprise

Original Work

This model is based on HSE-asavchenko/face-emotion-recognition. Weights converted and exported by facetorch.

Usage

import torch

# Load — no model class needed
ep = torch.export.load("model.pt2")
model = ep.module()

# Inference
x = torch.randn(1, 3, 224, 224)  # adjust size per model
output = model(x)

Or via facetorch:

from facetorch import FaceAnalyzer
from omegaconf import OmegaConf

cfg = OmegaConf.load("conf/config.yaml")
analyzer = FaceAnalyzer(cfg.analyzer)
response = analyzer.run(path_image="face.jpg")