请问一下这个模块在分割的时候如何设置dct_h和dct_w?
nnoyuwan opened this issue · comments
Thomas Gundersen commented
输入要经过下采样和上采样,输入图片也就需要缩小和放大好几次,那我是不是要在__init__就把input size传进去,有什么办法能实时获取输入大小吗?
我有尝试在forward函数里获取x的大小,但后面训练很慢,我就意识到dct应该需要在搭建网络时预生成。
cfzd commented
Thomas Gundersen commented
谢谢