langmanbusi/Semantic-Aware-Low-Light-Image-Enhancement

代码

Closed this issue · 2 comments

您好,我想问一下这个输入的不应该是图像特征和语义特征嘛,为什么直接就是96,32两个数字呀。
if seg:
self.fuseblk_1 = TransformerBlock(96, 32)
self.fuseblk_2 = TransformerBlock(48, 16)
self.fuseblk_3 = TransformerBlock(59, 16)

你好,
这里是模型class的init部分,模型需要用到的各网络层在这里定义,TransformerBlock在这里实例化定义为self.fuseblk,其中96和32是TransformerBlock的init参数,可以去对应的文件里详细看看


self.fuseblk的调用在forward函数里
f_s4 = self.fuseblk_1(f_s4, seg_fea[1])

可以再多看看代码,熟悉一下定义和调用,祝好^^

谢谢您了