argmax won’t work using a single channel, since this single channel will always contain the maximal value, so you’ll get a prediction containing all zeros.
However, the threshold method should work.
Could you apply a threshold of 0.5 and try it again?