win电脑TensorFlow训练时，遇到Invalid argument: Expected image (JPEG, PNG, or GIF), got unknown format的解决办法

HNU_刘yuan

于 2020-08-08 18:21:53 发布

阅读量2.6k

点赞数 4

分类专栏： python 文章标签： python tensorflow 深度学习

本文链接：https://blog.csdn.net/qq_40608730/article/details/107883641

版权

python 专栏收录该内容

36 篇文章

订阅专栏

本文详细解析了在使用tfrecord进行网络训练时遇到的“Expected image (JPEG, PNG, or GIF)”错误，揭示了错误根源在于图片文件夹中混有非图片格式文件，并提供了具体的排查与解决步骤，包括如何通过Python脚本定位并移除问题文件。

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

解决办法

我在生成tfrecord文件后，训练网络时遇到
Invalid argument: Expected image (JPEG, PNG, or GIF), got unknown format starting with ‘BM\356\243,\000\000\000\000\0006\000\000\000(\000’
从错误信息中可以看出，是由于图片文件夹中存在非图片文件，根据后面的信息可知我的是bm文件，这个不一定相同，根据自己的情况而来。

接着我们可以通过编写程序遍历所有文件找到：

import os
dir = 'image_path/'
for i, filename in enumerate(os.listdir(dir)):
    filename = dir + filename
    with open(filename, 'rb') as imageFile:
        if imageFile.read().startswith(b'xxx'):
            print(f"{i}: {filename} - found!")