python - Tensorflow Lite，图像大小零误差

Question

其实我的问题很简单。我想在tensorflow lite模型中使用我自己的数据。所以，我写了这行代码:

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)

另外，这是我遇到的错误:

  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

Answer 1

当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.