huawei-noah / Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.
4.02k stars 706 forks source link

我应该如何将GhostNetV2加入到yolov5中去? #222

Open nortorious opened 1 year ago

nortorious commented 1 year ago

如何修改成 C3GhostV2的模块?

nortorious commented 1 year ago

image image ghostnetV2中的bottleneckV2有三个通道数,而V1版本的bottleneck只有两个通道数,我想知道mid_chs是什么

Chy-ang commented 1 year ago

您好,请问您的ghostv2_pytorch/train.py中有mix_up函数吗?我看作者是从timm.data导出的,但是我这里爆红显示没有这个函数,版本和作者提供的是一致的

nortorious commented 1 year ago

我是在https://github.com/huawei-noah/Efficient-AI-Backbones/blob/master/ghostnetv2_pytorch/model/ghostnetv2_torch.py这个里面复制的GhostBottleneckV2模块,其他的没怎么弄