pytorch+numpy+PIL将dat文件读取为图像

Posted 2023-03-03 小明250

import os
import torch
import numpy as np
from PIL import Image
path = "D:/OAdata/Camera_3"
datanames = os.listdir(path)
fullpath = path + '/' + datanames[100]
print(fullpath)
data1 = np.fromfile(fullpath, dtype=np.uint8)
x = torch.from_numpy(data1)
d2 = x.reshape(1024, 1280)
print(d2)
d2 = d2.detach().numpy()
img = Image.fromarray(d2)
img.show()