如何将yolov5结合ghostnet

bubbliiiing / yolov5-pytorch

这是一个YoloV5-pytorch的源码，可以用于训练自己的模型。

GNU General Public License v3.0

1.04k stars 175 forks source link

Open DESMONDLUCY opened 2 years ago

DESMONDLUCY commented 2 years ago

导师你好，我在看了你的yolov4结合ghostnet视频后，想尝试将ghostnet结合进入yolov5。想请问，除了需要导入相应的ghostnet.py以外，还需要对bottldneckCSP或者其他地方进行修改吗？

bubbliiiing commented 2 years ago

bottldneckCSP？不用吧，你只要通道conv一下就可以

DESMONDLUCY commented 2 years ago

抱歉我没太理解，能具体解释一下吗

bubbliiiing commented 2 years ago

你可以参考一下next_version分支，里面有替换主干的