search

Human9000 / nd-Mamba2-torch

Only implemented through torch: "bi - mamba2" , "vision- mamba2 -torch". support 1d/2d/3d/nd and support export by jit.script/onnx;
189 stars 6 forks source link

vssd模块如何修改输出层的尺寸 #18

Open BBAALLDDD opened 1 week ago

BBAALLDDD commented 1 week ago

现在vssd 模块 输入 （1， 3， 256，256） 会输出 （1，64， 64， 64） （1， 128， 32， 32） （1， 256， 16， 16) (1, 512, 8, 8)

我想修改为（1， 64， 128， 128） （1， 256， 64， 64） （1， 512， 32， 32） （1，1024， 16， 16） 该修改哪里

Human9000 commented 1 week ago

感谢你的提问，有以下两种方案， 第一种相对简单，你可以尝试将当前的结果通过上采样或者反卷积得到你的结果。 第二种较为复杂，你可以打印vssd的不同层的特征图，查看在不同的网络层是否有你需要的特征图尺寸，使用hook方法或return多个特征图的方法来指定你需要的尺寸导出。