TensorFlow Lite,图像大小为零错误

Tensorflow Lite, Image size is zero error(TensorFlow Lite,图像大小为零错误)

本文介绍了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.

这篇关于TensorFlow Lite,图像大小为零错误的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持编程学习网!

本文标题为:TensorFlow Lite,图像大小为零错误

基础教程推荐