Tensorflow Lite, Image size is zero error-CodePudding

Actually, my question is very simple. I would like to use my own data in tensorflow lite model. So, i wrote these line of codes:

root_path = r"C:\Users\90531\Desktop\dataset\b"
image_path = os.path.join(os.path.dirname(root_path), '1602854451425')

data = DataLoader.from_folder(image_path)

Also, this is the error that I encountered:

  File "C:\Users\90531\AppData\Roaming\Python\Python39\site-packages\tensorflow_examples\lite\model_maker\core\data_util\image_dataloader.py", line 73, in from_folder
    raise ValueError('Image size is zero')

ValueError: Image size is zero

CodePudding user response：

This happens when the Dataloader cannot infer the labels of your images. The images should be divided into subfolders according to the class they belong to:

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.