zhigangjiang/3D-Room-Layout-Estimation_LGT-Net

1"""2@date: 2021/8/43@description:4"""5import numpy as np6import torch7 8 9def tensor2np(t: torch.Tensor) -> np.array:10    if isinstance(t, torch.Tensor):11        if t.device == 'cpu':12            return t.detach().numpy()13        else:14            return t.detach().cpu().numpy()15    else:16        return t17 18 19def tensor2np_d(d: dict) -> dict:20    output = {}21    for k in d.keys():22        output[k] = tensor2np(d[k])23    return output24