使用TensorFlow Lite加载图片数据集时报Image size is zero错误如何解决

ValueError: Image size is zero报错的核心是DataLoader.from_folder()方法没有读取到任何符合要求的有效图片，常见触发原因如下：

• 路径配置错误：你代码里构造的image_path路径不存在，或者路径下没有符合要求的图片文件。你编写的root_path为r"C:\Users\90531\Desktop\dataset\b"，os.path.dirname(root_path)会提取到上层路径r"C:\Users\90531\Desktop\dataset"，拼接后的image_path为r"C:\Users\90531\Desktop\dataset\1602854451425"，首先要确认该路径真实存在。
• 目录结构不符合要求：from_folder要求传入的根路径下必须按分类创建子文件夹，每个子文件夹内存放对应类别的图片，不能直接把所有图片放在根路径下，否则方法会直接过滤根目录下的图片，统计到的有效图片数为0。
• 图片不符合读取要求：路径下的图片格式不被支持（比如罕见格式、损坏的图片、0字节的空图片），或者Python进程没有权限读取路径下的文件，都会导致有效图片数为0。

• 校验路径和目录结构：先打印image_path确认路径正确，再检查路径下的结构是否符合要求，示例合规结构如下：

1602854451425/
├─ 分类A/
│  ├─ 图片1.jpg
│  └─ 图片2.jpg
└─ 分类B/
   ├─ 图片3.png
   └─ 图片4.jpeg

如果你的图片直接放在1602854451425根目录下，必须先按分类整理到对应子文件夹中。

• 校验图片有效性：确认路径下的图片都是JPG、PNG等常见支持格式，没有损坏或者0字节的空文件，可手动打开几个图片确认可用性。
• 校验读取权限：执行代码print(os.listdir(image_path))，如果能正常打印路径下的子文件夹/文件列表说明权限正常，如果报错就以管理员身份运行Python脚本，或者把数据集移动到不需要管理员权限的普通目录下再执行。

内容的提问来源于stack exchange，提问作者Berkay