PyTorch下的InsightFace: 预先训练模型的移植
5星
- 浏览量: 0
- 大小:None
- 文件类型：ZIP
简介：
本项目介绍了如何在PyTorch框架下成功移植预先训练好的InsightFace模型，为开发者提供面部识别应用的便捷部署方案。
Pytorch InsightFace 将来自预训练的ResNet模型移植到pytorch。
| 模型 | LFW（%） | CFP-FP（%） | AgeDB-30（%） | MegaFace（%） |
|------------|-----------|-------------|---------------|--------------|
| iresnet34 | 99.65 | 92.12 | 97.70 | 96.70 |
| iresnet50 | 99.80 | 92.74 | 97.76 | 97.64 |
| iresnet100 | 99.77 | 98.27 | 98.28 | 98.47 |
安装：
```
pip install git+https://github.com/nizhib/pytorch-insightface
```
用法：
```python
import torch
from imageio import imread
from torchvision import transforms
import insightface
embedder = insightface
```
全部评论 (0)
还没有任何评论哟~
