[FIXED] Tensorflow Lite, Image size is zero error

Issue

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

Solution

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.

Answered By - AloneTogether