search

wdhudiekou / UMF-CMGR

[IJCAI2022 Oral] Unsupervised Misaligned Infrared and Visible Image Fusion via Cross-Modality Image Generation and Registration
MIT License
170 stars 18 forks source link

关于test_reg_fusion.py中regnet的问题 #32

Open zhongniuniu opened 11 months ago

zhongniuniu commented 11 months ago

请问一下代码中96行 ir_pred, f_warp, flow, int_flow1, int_flow2, disp_pred = RegNet(it, ir)配准网络只有两个输入会导致报错，这如何解决？ 我看deformable_net中输入的确需要三个值def forward(self, tgt, src, real,shape=None): 而且我输入三个值，ir_pred输出的灰度值全是0，不知道是不是因为这个导致的

zhongniuniu commented 11 months ago

总是报TypeError: forward() missing 1 required positional argument: 'real'