TensorFlow-ESPCN
TensorFlow-ESPCN copied to clipboard
Problem with the _phase_shift() function
I suppose there's a little bug in the _phase_shift
function, cause if bsize
is 1 then after tf.squeeze(x)
the batch dimension would also disappear.
Maybe tf.squeeze(x, 1)
instead?