Segmentation result

[yahoo0742]

Hi @carrierlxk, I believe the output of the model has 2 tensors in the shape of (N_batch, 1_channel, H, W) respectively. In the code below https://github.com/carrierlxk/COSNet/blob/549109db0d60b69fd4f70b400bd48a12d6c83ea7/test_coattention_conf.py#L165

It only summed prediction for the first input of a batch data[0,0], and in the end for a batch, there is only one masks generated even though there are more than one input.