search

Zhaozixiang1228 / MMIF-CDDFuse

[CVPR 2023] Official implementation for "CDDFuse: Correlation-Driven Dual-Branch Feature Decomposition for Multi-Modality Image Fusion."
349 stars 34 forks source link

关于CNN Encoder的输入特征维度 #6

Closed Liujian1997 closed 1 year ago

Liujian1997 commented 1 year ago

您好！

经过Restormer后的特征图应该是（B，H*W，dim），想问一下是通过transpose(1, 2).reshape(B, dim, H, W)变成与CNN相似的特征图然后输入到CNN Encoder中吗？

Zhaozixiang1228 commented 1 year ago

The dim of feature output by Restormer is B,C,H,W.