关于用Ghostnetv2 替换resnet

huawei-noah / Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.

3.91k stars 695 forks source link

Open dazaier opened 3 months ago

dazaier commented 3 months ago

您好！我现在的网络用的resnet34 做的图像的特征提取，如果我现在想用ghostnetv2 替换掉resnet34进行特征提取，我能直接加载您的预训练模型吗？

iamhankai commented 3 months ago

可以

dazaier commented 3 months ago

谢谢，我看还有g_ghosnet_regnet,如果我更在乎模型在GPU上的显存占用和运行时间，是不是用g_ghost比ghostnetv2更好呀！

iamhankai commented 3 months ago

你都测测看