TensorFlow Lite，图像大小为零错误

问题描述

实际上，我的问题很简单。我想在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.