xujiashuai/timm-cv_resnet34_face-attribute-recognition_fairface
模型介绍文件和版本Pull Requests讨论分析
下载使用量0

cv_resnet34_face-attribute-recognition_fairface - 昇腾 NPU 适配

1. 模型简介

ResNet34 人脸属性识别模型，基于 FairFace 数据集训练，可实现 18 类属性分类。

  • 原始模型: iic/cv_resnet34_face-attribute-recognition_fairface
  • 框架: PyTorch (torchvision)
  • 任务: 人脸属性识别（18 类）

2. 昇腾 NPU 适配结果

指标值
余弦相似度1.000000
平均延迟3.41 ms
推理精度float16
设备Ascend 910B4

3. 快速使用

export ASCEND_HOME_PATH=/usr/local/Ascend/cann
export LD_LIBRARY_PATH=/usr/local/Ascend/cann/lib64:$LD_LIBRARY_PATH
python inference.py --device npu:0 --dtype float16

4. 推理输出证据

模型: iic/cv_resnet34_face-attribute-recognition_fairface
设备: npu:0
精度: float16
------------------------------------------------------------
  input: shape=torch.Size([1, 3, 224, 224])
  logits: shape=torch.Size([1, 18])
  CPU pred: 12, NPU pred: 12

  Cosine Similarity (logits): 1.000000

✓ 推理完成
  平均延迟: 3.41 ms

5. CPU 与 NPU 精度对比

指标CPU (float32)NPU (float16)误差
余弦相似度基准1.0000000%
输出维度[1,18][1,18]一致
预测类别1212一致

6. 智能体技能

本适配由 batch-adapter 自动完成。