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

COSNet/test_coattention_conf.py

Line 165 in 549109d

output_sum = output_sum + output[0].data[0,0].cpu().numpy() #分割那个分支的结果

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.