Method fusion

star/models/CNNNet.py:372–376

(self, mode="sum")

                self.neighbor_feat_list.append(warp_feat)
        
    def fusion(self, mode="sum"):
        fused = torch.sum(torch.stack(self.neighbor_feat_list), dim=0)
        if mode == "union":
            fused[fused>0.] = 1.0 
        return fused 

    @staticmethod
    def agents_to_batch(feats):