caiyuanhao1998 / MST

A toolbox for spectral compressive imaging reconstruction including MST (CVPR 2022), CST (ECCV 2022), DAUHST (NeurIPS 2022), BiSCI (NeurIPS 2023), HDNet (CVPR 2022), MST++ (CVPRW 2022), etc.
MIT License
571 stars 71 forks source link

测试代码问题 #45

Closed dongruoyu1 closed 7 months ago

dongruoyu1 commented 7 months ago

image 学长你好,我在上个问题中,将num_worker从8修改为0之后可以成功训练模型,将训练好的模型进行测试,出现了以下的bug,在csdn上也找不到解决办法,所以想来请教一下您,说是参数有问题,但是您的代码我也没怎么修改过呀。

caiyuanhao1998 commented 7 months ago

用 ubuntu 跑

dongruoyu1 commented 7 months ago

用 ubuntu 跑

谢谢您的回复,我试试看。 还有一个小问题,在阅读代码的时候,看到了mask_3d_shift_s,mask是初始二维掩码660×660,mask_3d是mask经过28次复制后得到的660×660×28,而mask_3d_shift是经过平移后得到的660×714×28,那么mask_3d_shift_s是将mask_3d_shift每一个像素平方后,在图片维度相加,得到的尺寸为660×714,我不太理解mask_3d_shift_s为什么要通过平方操作获得,以及mask_3d_shift_s的意义是什么? image

dw0127 commented 5 months ago

用 ubuntu 跑

谢谢您的回复,我试试看。 还有一个小问题,在阅读代码的时候,看到了mask_3d_shift_s,mask是初始二维掩码660×660,mask_3d是mask经过28次复制后得到的660×660×28,而mask_3d_shift是经过平移后得到的660×714×28,那么mask_3d_shift_s是将mask_3d_shift每一个像素平方后,在图片维度相加,得到的尺寸为660×714,我不太理解mask_3d_shift_s为什么要通过平方操作获得,以及mask_3d_shift_s的意义是什么? image

请问用ubuntu 跑后结果如何,我目前也遇到这个问题。谢谢