Function numpy2pytorch

freesplatter/utils/infer_util.py:37–40

(imgs)

@torch.inference_mode()
def numpy2pytorch(imgs):
    h = torch.from_numpy(np.stack(imgs, axis=0)).float() / 255.0 * 2.0 - 1.0
    h = h.movedim(-1, 1)
    return h


@torch.inference_mode()