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,图像大小为零错误的教程就到这里就结束了,希望趣模板源码网找到的这篇技术文章能帮助到大家,更多技术教程可以在站内搜索。