利用pytorch训练好的模型测试单张图片
pytorch网络输入图片的格式是[B,C,H,W],分别为批大小(batchsize),图片通道数(channel),图片高(height),图片宽(width)。图片读取方式主要有两种:(1)通过PIL进行读取;(2)通过opencv进行读取。分别进行介绍。(1)通过PIL进行读取:通过PIL的Image读取的图片是一个图片对象,可以进行裁剪翻转等torchvision.transforms变换。torchvision.transforms可以对图像对象进行一系列裁剪、翻转等转换操.
原创
2020-10-11 16:38:04 ·
18602 阅读 ·
15 评论