Cleanup excessive axis swaps

[tpet93]

When converting between tensors, numpy arrays and through different stages of the workflow the axis are swapped and flipped an excessive amount.

for example some parts of the workflow require [batch,classes,x,y] and others require [batch,x,y,classes]. there are also issues when running single class, band or batch's with the need to unsqueeze.

in some cases it appears converting between numpy and tensor flips the image.