bubbliiiing / yolov4-tiny-pytorch

这是一个YoloV4-tiny-pytorch的源码,可以用于训练自己的模型。
MIT License
787 stars 184 forks source link

你好,关于attention.py #36

Open LaoGanGit opened 3 years ago

LaoGanGit commented 3 years ago

你好,我打算自己在attention.py里加一些新的注意机制在用train里的phi调用,想请教在train里的phi调用SE,CBAM的逻辑走向是如何的,我应该怎么调用自己的新加的关注机制

bubbliiiing commented 3 years ago

我感觉这个attention的部分的代码很简单了已经。我不知道解释什么……