TensorFlow Lite,图像大小为零错误

本教程将介绍TensorFlow Lite,图像大小为零错误的处理方法,这篇教程是从别的地方看到的,然后加了一些国外程序员的疑问与解答,希望能对你有所帮助,好了,下面开始学习吧。

TensorFlow Lite,图像大小为零错误 教程 第1张

问题描述

实际上,我的问题很简单。我想在TensorFlow Lite模型中使用我自己的数据。因此,我编写了以下代码行:

root_path = r"C:Users90531Desktopdataset"
image_path = os.path.join(os.path.dirname(root_path), '1602854451425')

data = DataLoader.from_folder(image_path)

另外,这是我遇到的错误:

  File "C:Users90531AppDataRoamingPythonPython39site-packages	ensorflow_exampleslitemodel_makercoredata_utilimage_dataloader.py", line 73, in from_folder
 raise ValueError('Image size is zero')

ValueError: Image size is zero

推荐答案

Dataloader无法推断出您的图像的标签时会发生这种情况。应根据图像所属的类别将子文件夹划分为子文件夹:

from tflite_model_maker.image_classifier import DataLoader
import seedir as sd

image_path = '/content/images'
sd.seedir(image_path, style='spaces', indent=4, anystart='- ')
data = DataLoader.from_folder(image_path)
- images/
 - class1/
  - result_image.png
 - class2/
  - result_image1.png

INFO:tensorflow:Load image with size: 2, num_label: 2, labels: class1, class2.

好了关于TensorFlow Lite,图像大小为零错误的教程就到这里就结束了,希望趣模板源码网找到的这篇技术文章能帮助到大家,更多技术教程可以在站内搜索。