Flow is reshaped to correct remapping map

The flow cannot be used directly to remap the images. Hence a method was
introduced to calculate the corresponding mapping.