首页import torch import numpy as np from torchvision import datasets, models, transforms from torchvision import transforms from PIL import Image transform = transforms.Compose([ transforms.Resize((28, 28), interpolation=2), transforms.ToTensor() ] ) a=np.load('./original_crystal_2d_graphs/mp-170_copy1_opt.vasp_subst_O-Se_V-Bi.npy') img = a[0:3, :] #数组形式 im = Image.fromarray(img) # img = Image.open(im) img_ = transform(im).unsqueeze(0) #拓展维度 img_.show()

import torch import numpy as np from torchvision import datasets, models, transforms from torchvision import transforms from PIL import Image transform = transforms.Compose([ transforms.Resize((28, 28), interpolation=2), transforms.ToTensor() ] ) a=np.load('./original_crystal_2d_graphs/mp-170_copy1_opt.vasp_subst_O-Se_V-Bi.npy') img = a[0:3, :] #数组形式 im = Image.fromarray(img) # img = Image.open(im) img_ = transform(im).unsqueeze(0) #拓展维度 img_.show()

时间: 2023-06-07 22:08:23 浏览: 233

关于PyTorch源码解读之torchvision.models

5星 · 资源好评率100%

PyTorch框架中有一个非常重要且好用的包：torchvision，该包主要由3个子包组成，分别是：torchvision.datasets、torchvision.models、torchvision.transforms。这3个子包的具体介绍可以参考官网： http://pytorch.org/docs/master/torchvision/index.html。具体代码可以参考github： https://github.com/pytorch/vision/tree/master/torchvision。这篇博客介绍torchvision.models。torchvision.

这段代码的作用是，导入了PyTorch、NumPy和torchvision等工具库，并指定了一组数据转换操作（修改为28x28尺寸并转换为张量格式）。然后，使用NumPy库读取了一个二维晶体图形的数据，并取出前三行作为图像数据。

阅读全文