花分类
使用mobilenet v2对102种花朵进行分类并进行转移学习。
使用Costom数据集对花朵图像进行分类
下载资料
在该项目中,使用了 。 您可以下载和。 数据集为tar文件形式,一旦提取,所有图像均位于jpg文件夹内。 标签文件包含一个数组,其索引是图像的编号,而索引处的编号是该图像所属类的编号。
Keras数据生成器
此处,keras图像生成器用于将数据馈送到神经网络。 数据的结构应采用以下方式: 。
在上面的图像中,training_images是包含多个目录的子文件夹,并且每个目录包含属于一个垂直类的图像。
数据准备
由于keras数据生成器要求训练数据采用垂直结构,因此我们需要根据需要准备数据集。 要准备数据集,请参考,将以上述格式转换数据。
在文件中,您需要更改以下变量:
TAR_FILE_PATH =''#提供下载的tar文件的路径EXTRACT_TO_PA
2021-10-25 19:47:21
10.06MB
Python
1