200字范文,内容丰富有趣,生活中的好帮手!
200字范文 > 怎么利用pytorch训练好的模型测试单张图片

怎么利用pytorch训练好的模型测试单张图片

时间:2021-07-10 21:51:56

相关推荐

怎么利用pytorch训练好的模型测试单张图片

pytorch网络输入图片的格式是[B,C,H,W],分别为batch(每批送入网络的图片数量),图片通道数,图片高,图片宽。

torchvision.transforms可以对图像对象进行一系列裁剪、翻转等转换操作,其中也包括转换为tensor张量。(transforms.ToTensor())

而opencv读取的是ndarray格式,不能进行torchvision.transforms变换。

通过PIL的Image读取的图片是一个图片对象,可以进行裁剪翻转等torchvision.transforms变换。

np.ndarray与PIL.Image对象相互转换:

numpy.array(img) img对象转化为np数组Image.fromarray(arr) np数组转化为img对象

测试代码如下:

from PIL import Imageimport torchvision.transforms as Tfrom models.mobilenet_master2 import MobileNet#导入自己定义的网络模型from torch.autograd import Variable as Vimport torch as ttrans=pose([transforms.Scale(256),transforms.CenterCrop(224),transforms.ToTensor(),T.Normalize(mean=[0.485, 0.456, 0.406],

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。