ValueError: cannot reshape array of size xx into shape (x,x,x)

最新推荐文章于 2023-09-04 10:47:49 发布

白刃的阿白

最新推荐文章于 2023-09-04 10:47:49 发布

阅读量7.5k

点赞数 4

分类专栏： bug记录文章标签：计算机视觉 python

本文链接：https://blog.youkuaiyun.com/qq_31930361/article/details/121381741

版权

在使用CTPN训练时，遇到ValueError：无法将大小为571428的数组重塑为(3,351,407)。原因是一张图像的通道、宽度和高度的乘积不匹配。经检查，该图像是一个4通道的PNG文件，误标为.jpg。解决方案是将图像转换为RGB三通道图像。" 117968890,7656398,深入理解LinkedList：面试必备知识,"['数据结构', '链表', 'java', '面试', '队列']

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

ValueError: cannot reshape array of size 571428 into shape (3,351,407)

在训练CTPN的时候，数据集处理的 cv2.dnn.blobFromImage 之后的reshape报的这个错。原因是有一张图像它的通道数乘以宽和高等于571428，不等于3 * 351 * 407，因此不能reshape到（3，351，407）。
算了一下 571428 = 4 * 351 * 407 ，说明这个图莫名其妙地是个4通道的图像。然后断点找了一下这张图，发现它竟然是一个png格式的后缀名jpg的假jpg，多了一个通道。因为png是RGBA，jpg是RGB。
解决：进行图像转换，把它转换为RGB的三通道图。在这

    def __getitem__(self, idx, scale=600, maxScale=900):
        img_name = self.img_names[idx]
        img_path = os.path.join(self.datadir, img_name)
        img = Image.open(img_path)
        img, rate = resize_img(img