tomas-gajarsky/facetorch-au-opengraph

Facial Action Unit Detection (41 AUs)

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

Model Details

Task	Facial Action Unit Detection (41 AUs)
Architecture	Swin Transformer Base (timm) + GNN head
Format	torch.export (.pt2) — no model source code needed
Dynamic shapes	Batch dimension is dynamic (1-64)
Input	224x224 face crop
Output	41-dim scores (27 main + 14 sub action units)

Original Work

This model is based on lingjivoo/OpenGraphAU, CVI-SZU/ME-GraphAU. 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")