Views
No views yet
| Metric | Baseline (pretrained) | Trained (ArcFace) |
|---|---|---|
| Precision@10 | 0.696 | 0.850 |
| mAP@10 | 0.790 | 0.901 |
1import torch
2from PIL import Image
3from torchvision import transforms
4from modeling import load_model
5
6model, cfg = load_model('.', device='cuda' if torch.cuda.is_available() else 'cpu')
7tfm = transforms.Compose([
8 transforms.Resize((cfg['image_size'], cfg['image_size'])),
9 transforms.ToTensor(),
10 transforms.Normalize(cfg['normalize_mean'], cfg['normalize_std']),
11])
12
13@torch.no_grad()
14def embed(path):
15 x = tfm(Image.open(path).convert('RGB')).unsqueeze(0)
16 return model(x.to(next(model.parameters()).device)).cpu().numpy()