tomas-gajarsky/facetorch-embed-resnet-50

Face Embedding

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

Model Details

Task	Face Embedding
Architecture	ResNet-50 (SENet variant)
Format	torch.export (.pt2) — no model source code needed
Dynamic shapes	Batch dimension is dynamic (1-64)
Input	244x244 face crop
Output	2048-dim embedding vector

Original Work

This model is based on cydonia999/VGGFace2-pytorch. 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")