使用 Colab 训练图像分类模型

我按照模板修改了链接,但它不起作用https://colab.research.google.com/github/tensorflow/tensorflow/blob/master/tensorflow/lite/g3doc/tutorials/model_maker_image_classification.ipynb#scrollTo=3jz5x0JoskPv

这是我的数据集https://firebasestorage.googleapis.com/v0/b/lol-fypproject.appspot.com/o/lol.tgz?alt=media&token=d07b81bd-442f-4ebe-920e-3772598fbb20

原始代码

image_path = tf.keras.utils.get_file(  'flower_photos.tgz',  'https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz',  extract=True) image_path = os.path.join(os.path.dirname(image_path), 'flower_photos')

我修改为

image_path = tf.keras.utils.get_file(  'lol.tgz',  'https://firebasestorage.googleapis.com/v0/b/lol-fypproject.appspot.com/o/lol.tgz?alt=media&token=d07b81bd-442f-4ebe-920e-3772598fbb20',  extract=True)image_path = os.path.join(os.path.dirname(image_path), 'lol')

这行代码出错了,错误信息如下

data = ImageClassifierDataLoader.from_folder(image_path)---------------------------------------------------------------------------ValueError                                Traceback (most recent call last)<ipython-input-15-a5e7646aca55> in <module>()----> 1 data = ImageClassifierDataLoader.from_folder(image_path)  2 train_data, test_data = data.split(0.9)/usr/local/lib/python3.7/dist- packages/tensorflow_examples/lite/model_maker/core/data_util/image_dataloader.py  in  from_folder(cls, filename, shuffle) 69     all_image_size = len(all_image_paths) 70     if all_image_size == 0: ---> 71       raise ValueError('Image size is zero') 72  73     if shuffle: ValueError: Image size is zero

回答:

我已经找到了问题所在

压缩文件的路径结构与示例不符

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注