目录
pytorch做训练时,因为pytorch没有自带的读取图片的方法,只能通过PIL和OpenCV两个包来读取。有时候为了方便加速训练,将数据和模型都转换到GPU上,所以我们经常会将图片数据转换为tensor格式。
但是有很多时候我们会需要展示训练图片的的结果。所以难免会涉及到将Tensor数据转换回方便展示的PIL和opencv(numpy)的格式。
1.python中的几种图片格式
- PIL:使用python自带图像处理库读取出来的图片格式,颜色通道是RGB,读取后数据大小为[C, H, W]。取值范围一般是[0,255]
- nump