multispectral-object-detection icon indicating copy to clipboard operation
multispectral-object-detection copied to clipboard

关于ADD2模型,为什么在add命令上会有不同,难道不是相同位置元素相加吗?请问为什么这样设置add

Open leileilei2000 opened this issue 1 year ago • 2 comments

class Add2(nn.Module): # x + transformer[0] or x + transformer[1] def init(self, c1, index): super().init() self.index = index

def forward(self, x):
    if self.index == 0:
        return torch.add(x[0], x[1][0])
    elif self.index == 1:
        return torch.add(x[0], x[1][1])
    # return torch.add(x[0], x[1])

leileilei2000 avatar Apr 10 '24 03:04 leileilei2000

一个是红外加到可见光,一个是可见光加到红外,具体你要看yolo_test.py文件里面的计算流程

Yipzcc avatar Apr 11 '24 10:04 Yipzcc

一个是红外加到可见光,一个是可见光加到红外,具体你要看yolo_test.py文件里面的计算流程

请问为什么我的输出没有任何可视化的检测结果,train和test都没有图像可视化检测结果

leileilei2000 avatar Apr 23 '24 12:04 leileilei2000