tensorflow tfrecords处理图片踩坑

最近用tensorflow做一个表情识别的应用,数据集全部都是自己收集的,随后制作为128 X 128 X 3的tfrecords。在输入数据时总是报错,说要求输入为49152(128 X 128 X 3),而输入为65536(128 X 128 X 4)。
我就很纳闷为什么会有这样的事情,原来tensorflow默认处理的图片位深度为24,而有些图片的位深度为32,这就引起了这种错误

你可能感兴趣的:(tensorflow)