基于mobileNet实现狗的品种分类(迁移学习)

基于mobileNet实现狗的品种分类

• 一、数据集介绍
• 二、实战
• • 2.1 数据预处理
  • • 2.1.1 第1步：对labels标签的预处理
    • 2.1.2 第2步：对图片数据的预处理
  • 2.2 迁移学习、模型微调
  • 2.3 模型训练、验证、测试
  • 2.4 模型预测
  • 2.5 模型保存

一、数据集介绍

数据集下载地址：https://www.kaggle.com/datasets/carlosmiao/dogbreedidentification

狗的品种共120种

10222张训练图片

train和test文件夹里面是图片

labels.csv种记录着图片对应的标签值。