Eu tentei:

test_image = tf.convert_to_tensor(img, dtype=tf.float32)

Aparece o seguinte erro:

ValueError: Tensor conversion requested dtype float32 for Tensor with dtype int64: 'Tensor("test/ArgMax:0", shape=TensorShape([Dimension(None)]), dtype=int64)'