图像模式错误：非RGB

最新推荐文章于 2023-04-03 17:25:31 发布

摸鱼时刻

最新推荐文章于 2023-04-03 17:25:31 发布

阅读量2.2k

点赞数 1

分类专栏：人工智能

本文链接：https://blog.csdn.net/u014657795/article/details/81346996

版权

人工智能专栏收录该内容

4 篇文章 1 订阅

订阅专栏

用caffe训练模型的时候会遇到如下两种错误：

OpenCV Error: Assertion failed ((scn == 3 || scn == 4) && (depth == CV_8U || depth == CV_32F)) in cvtColor, file /ssda/software/opencv-3.1.0/modules/imgproc/src/color.cpp, line 8141
terminate called after throwing an instance of 'cv::Exception'

或者：

annotated_data_layer.cpp:205] Check failed: std::equal(top_shape.begin() + 1, top_shape.begin() + 4, shape.begin() + 1)

原因：图片数据集中存在非RGB模式的图片，如灰度图（L）解决方法：

去掉非RGB图片
python将图片转换为RGB格式
代码如下：

from PIL import Image
import os
path = "./JPEGImages/"
for file_name in os.listdir(path):
    img =Image.open(path + file_name)
    if img.mode != "RGB" :
        print(file_name)
        img_rgb = img.convert("RGB")
        os.remove(path + file_name)
        img_rgb.save(path + file_name)

参考如下：
[1] https://blog.csdn.net/asukasmallriver/article/details/72954788
[2] https://blog.csdn.net/zj360202/article/details/78539040