n06_pytorch_utils.py 文件源码

def img_to_tensor(img, mean=0, std=1.):
    img = img.astype(np.float32)
    img = (img - mean) / std
    img = img.transpose((2, 0, 1))
    tensor = torch.from_numpy(img)  ##.float()
    return tensor


## for debug