图像计数为零

Question

我正在尝试使用 tensorflow lite 为 android 应用创建图像分类模型：https://www.tensorflow.org/tutorials/images/classification 但是，我正在使用我的本地目录。

这是我的代码：

import matplotlib.pyplot as plt
import numpy as np
import os
import PIL
import tensorflow as tf

from tensorflow import keras
from tensorflow.keras import layers
from tensorflow.keras.models import Sequential

import pathlib
data_dir = pathlib.Path('C:/Users/aalroumi/Documents/MLKIT/V')

image_count = len(list(data_dir.glob('*/*.JPEG')))
print(image_count)

这是输出图像：

![][1]

这是我本地目录的图像：

那么为什么我得到的图像大小为零？

我错过了什么吗？我做对了。

因为我正在尝试 files.download('model.tflite')

图像大小为零错误：

谢谢

Answer 1

您似乎直接在 data_dir 中有图像，因此您应该使用 *.JPEG 将它们放入 glob()

但是 from_folder(data_dir) 需要子文件夹中的图像，其名称将用作图像的 labels。

即

data_dir/cat/small_cat.JPEG
data_dir/cat/big_cat.JPEG
# ...
data_dir/dog/white_dog.JPEG
data_dir/dog/black_dog.JPEG

但是您没有子文件夹，因此无法在子文件夹中找到图片。
它不直接在 data_dir

中计算图像

图像计数为零

Image count is zero

python

tensorflow-lite

google-codelab