当前位置:   article > 正文

关于transform转变cv图片颜色不符的问题_cv::transform

cv::transform

学习pytorch过程中,发现当cv图片经过ToTensor后颜色发生变化,黄色会变成蓝色,蓝色会变成黄色

 

 这是由于cv2.imread读取图片的时候,是通过BGR形式读取图片,而ToTensor默认以RGB模式,这样转变后会导致颜色完全相反。

正确做法应当是将imread之后的图片修改成RGB模式:

  1. img=cv2.imread(img_path)
  2. img_rgb=cv2.cvtColor(img,cv2.COLOR_BGR2RGB)#格式转换,
  3. writer = SummaryWriter("logs")
  4. tensor_trans=transforms.ToTensor()
  5. tensor_img=tensor_trans(img_rgb)
  6. writer.add_image("bee", tensor_img,3,dataformats="CHW")

 这样最后得到的图片颜色就是正常的

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/513908
推荐阅读
  

闽ICP备14008679号