search

DanJun6737 / TransFace

[ICCV 2023] TransFace: Calibrating Transformer Training for Face Recognition from a Data-Centric Perspective
52 stars 7 forks source link

模型推理cpu占用高，gpu几乎不占用 #11

Closed MisakaMikoto-o closed 1 month ago

MisakaMikoto-o commented 1 month ago

大佬你好，我在调用这个模型推理的时候，发现cpu占用很高，但是gpu占用却几乎不升高，但是加载模型后显存占用确实变高了，至少是用到了显卡的，不知道是不是加载模型的时候没有完全加载到显卡中，调用和推理代码不是完全用的给出的demo，我把人脸检测和对齐部分用其他方法实现了，能帮我看看代码吗，谢谢

MisakaMikoto-o commented 1 month ago

https://colab.research.google.com/drive/1GYDmLzLvkhg6qwJul-2wwiLv4mWczVxQ?usp=sharing

MisakaMikoto-o commented 1 month ago

这个不能直接运行，缺少的人脸检测模型和人脸分割对齐库如果您需要的话我可以发给您

DanJun6737 commented 1 month ago

Hi~@MisakaMikoto-o

我简单看了一下，听您这描述 “调用和推理代码不是完全用的给出的demo，我把人脸检测和对齐部分用其他方法实现了”，有可能是这里出了问题，怀疑是跟TransFace那块不兼容。TransFace整个框架是基于InsightFace实现的，如果你想改人脸检测和对齐部分，最好基于InsightFace来改吧，这样可能会避免这些bug。

MisakaMikoto-o commented 4 weeks ago

嗯，找到原因了，是onnxruntime-gpu版本和cudnn版本不兼容导致的，已修复

DanJun6737 commented 4 weeks ago

嗯，找到原因了，是onnxruntime-gpu版本和cudnn版本不兼容导致的，已修复

嗯嗯，解决了就好！