outputs.cpu().detach().numpy()[0,:,:,:].shape
import matplotlib.pyplot as pyplot pyplot.imshow(binary_mask)pyplot.show()
