search

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

Semantic-Aware LLIE. CVPR 2023
MIT License
189 stars 7 forks source link

代码 #25

Closed chenhuimei55 closed 2 months ago

chenhuimei55 commented 5 months ago

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

langmanbusi commented 5 months ago

你好， 这里是模型class的init部分，模型需要用到的各网络层在这里定义，TransformerBlock在这里实例化定义为self.fuseblk，其中96和32是TransformerBlock的init参数，可以去对应的文件里详细看看 https://github.com/langmanbusi/Semantic-Aware-Low-Light-Image-Enhancement/blob/a2ade551508ecc7ea48f56ee5e49f023ea534c77/DRBN_SKF/src/model/Fuseblock.py#L133 self.fuseblk的调用在forward函数里 https://github.com/langmanbusi/Semantic-Aware-Low-Light-Image-Enhancement/blob/a2ade551508ecc7ea48f56ee5e49f023ea534c77/DRBN_SKF/src/model/drbn.py#L173 可以再多看看代码，熟悉一下定义和调用，祝好^^

chenhuimei55 commented 5 months ago

谢谢您了