《An Image Patch is a Wave: Phase-Aware Vision MLP》结构图+个人做的验证实验

今天阅读了《An Image Patch is a Wave: Phase-Aware Vision MLP》这篇论文,根据代码绘制的它的结构图。如果有错误,还请指正。

Wave_MLP_B 2D图像任务参数量有点大。换成3D估计没的玩了。

参数量:

1)  我用他代码跑了一下2D的胃部分割实验,发现了提取不了任何特征,也就是你的Loss始终都不会降低的。可能是我的问题,也可能是他依赖于预训练模型的原因吧(狗头)。

2)我将他的 wave_block 放到我的模型中,发现他推理结果及其不稳定,甚至会有两个点的差距。在肝脏血管分割实验中,我得到推理结果Dice值在 69.62%-71.78% 波动。我认为是因为它里面用了cos和sin的原因,也就是它提出的创新点,因为这个模块是不会学习到参数的,就会导致你每次推理结果都会发生变化。

 

posted @ 2022-06-10 19:57  九叶草  阅读(290)  评论(0编辑  收藏  举报