multispectral-object-detection
multispectral-object-detection copied to clipboard
关于ADD2模型,为什么在add命令上会有不同,难道不是相同位置元素相加吗?请问为什么这样设置add
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])
一个是红外加到可见光,一个是可见光加到红外,具体你要看yolo_test.py文件里面的计算流程
一个是红外加到可见光,一个是可见光加到红外,具体你要看yolo_test.py文件里面的计算流程
请问为什么我的输出没有任何可视化的检测结果,train和test都没有图像可视化检测结果