pytorch读取一张图像进行分类预测需要注意的问题(opencv、PIL)
1.我们在读取图像时可能会执行如下代码:
from PIL import Image
I = Image.open('D:/lung_project/dog_breed/test/00a3edd22dc7859c487a64777fc8d093.jpg')
print(I.shape)
这样就会出现如下错误:
from PIL import Image
I = Image.open('D:/lung_project/dog_breed/test/00a3edd22dc7859c487a64777fc8d093.jpg'

本文探讨了在使用pytorch进行图像分类预测时,使用opencv和PIL读取图像可能出现的问题。opencv读取的图像为BGR格式,而PIL则是RGB格式,这可能导致错误。正确处理图像格式对于避免预测错误至关重要。
最低0.47元/天 解锁文章
350

被折叠的 条评论
为什么被折叠?



