[深度学习] 彩色图片的通道排布问题

当我们用图像相关模型推理的时候, 需要考虑图像的通道问题, 特别是在部署的时候,不然会遇到数据不能对应的问题

  • opencv 的Mat 
    • 默认imread读取的时候, 内存的排布顺序是HWC, 可以用过Mat的data指针访问(访问时注意,data指针一定是char*, 如果你是float型矩阵, 需要强转一下)
    • 彩色通道的问题,也就是C的顺序,opencv默认是BGR,但一般我们是RGB,可以用converTo() API转一下.
  • PIL读取
    • 读取的图片内存排布是HWC
    • 彩色通道是RGB
  • Tensor( torch : transform.ToTensor())
    • 是将输入的数据shape W,H,C ——> C,H,W
    • 将所有数除以255,将数据归一化到【0,1】

    • 这个不涉及彩色通道,原图读取时时什么通道就是什么通道

猜你喜欢

转载自blog.csdn.net/qq_31638535/article/details/128663501