首页 > 分享 > 神经网络实现猫的种类识别

神经网络实现猫的种类识别

神经网络实现猫的种类识别

最新推荐文章于 2024-09-15 15:58:40 发布

yeler082 于 2019-01-12 19:59:11 发布

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。

神经网络实现猫的种类识别

代码地址:https://github.com/yeLer/cat_kind
在这里插入图片描述

实现的主要部分有 简单的网络结构在猫的种类识别上的应用,测试准确率0.832使用resNet50在猫的种类识别上的应用,测试准确率0.866使用数据增强解决过拟合问题,并提高测试准确率,测试准确率0.999 项目文件夹说明

------ cat_kind 项目文件名 |----cat_data_resNet50 存放用于resNet网络训练的数据 |----train 训练数据 |----test 测试数据 |----cat_kind_data 存放用于简单案例网络训练的数据(train.py) |----train 训练数据 |----test 测试数据 |----dataAug 数据增强后存放训练集的位置 |----models 保存训练模型的位置 |----predict_imgs 测试过程中使用的图片文件夹 |----resnet_example 使用resnet网络的文件夹 |----predict_on_resnet.py 预测的时候使用的文件 |----resnet50.py 定义resnet网络的文件 |----train_on_resnet.py 训练resnet的文件 |----source_images 数据集的原始文件夹 |----孟买猫 |----布偶猫 |----暹罗猫 |----英国短毛猫 |----tools 工具文件夹 |----data_aug.py 数据增强 |----prepare_data.py 数据预处理 |----predict.py 使用简单网络进行预测的文件 |----requirement.txt 环境依赖 |----train.py 使用简单网络进行训练的文件

1234567891011121314151617181920212223242526 step1 数据集预处理

使用文件:prepare_data.py

包括的操作有:

图片重命 在renameJPG()方法实现,重命名的意图有两个,第一是将图片都统一为一种图片类型,
第二个目的是给图片添加训练标签(当然也可以将标签写在文件当中,这里只是一种简便的
做法) 图片转换为统一的格式 图片重命名保存的过程中将图片统一格式 图片统一大小 该网络要求输入图片的大小为(100,100),所以运行时要修改out_img_size为(100,100),数据集保存位置save_dir为cat_kind_data 将图片按照一定的比例划分 按照比例9:1将数据划分为训练集和测试集,默认参数为0.9 step2 训练 使用文件:train.py,直接执行 step3 测试 使用文件:predict.py,直接执行 二 使用经典网络ResNet50实现猫的种类识别 由于resnet网络较为复杂,在少量的数据集情况下结果容易出现过拟合,所以首先进行数据增强的处理 step1 数据增强

在这里插入图片描述

直接运行data_aug.py,然后会在dataAug文件夹下生成四个对应猫种类的文件夹,相应的参数可以在data_aug.py调整

在这里插入图片描述

step2 数据集预处理

使用文件:prepare_data.py

包括的操作有:

图片重命名 在renameJPG()方法实现,重命名的意图有两个,第一是将图片都统一为一种图片类型,
第二个目的是给图片添加训练标签(当然也可以将标签写在文件当中,这里只是一种简便的
做法) 图片转换为统一的格式 图片重命名保存的过程中将图片统一格式 图片统一大小 在resNet中要求图片的大小在197——224之间,这里将图片的大小统一在224*224 将图片按照一定的比例划分 按照比例9:1将数据划分为训练集和测试集 step3 定义网络结构

网络结构采用Keras搭建好的resNet50,使用文件:resnet_example/resnet50.py

使用文件:resnet_example/resnet50.py,原始文件可从deep-learning-models
获取,为了使得该文件能适应训练加测试,我做了相应的修改

step4 训练 使用文件:resnet_example/train_on_resnet.py step5 测试 使用文件:resnet_example/predict_on_resnet.py

相关知识

基于卷积神经网络的宠物识别
基于卷积神经网络通过声音识别动物情绪的方法及系统
100基于卷积神经网络之鸟鸣识别鸟的种类
基于卷积神经网络的鸟类(声音和图片)识别项目
基于卷积神经网络通过声音识别动物情绪的方法及系统与流程
基于卷积神经网络的宠物识别 Pet Recognition Based on Convolutional Neural Network
详解pytorch实现猫狗识别98%附代码
基于深度神经网络的多模态情感识别
毕业设计:基于深度学习的宠物狗种类识别 人工智能 YOLO
【计算机科学】【2019.03】基于深度学习的动物识别

网址: 神经网络实现猫的种类识别 https://m.mcbbbk.com/newsview559442.html

所属分类:萌宠日常
上一篇: 最可爱的 13 种黑猫品种,每个
下一篇: 中华田园猫有什么种类(8种出名的