/TheCatLookLikeDog

这是一个深度学习的一个小项目,利用卷积神经网络识别猫狗图片

Primary LanguagePython

TheCatLookLikeDog

这是一个基于pytorch的卷积神经网络实现了猫狗图片区分的一个小项目

开发环境

python 3.9.7 pytorch 1.10.1

文档

net.py:卷积神经网络模型

train.py:用于训练及验证模型代码

split_data.py: 数据分流器,划分训练集和验证集

test1.py:用于测试模型

代码启动

猫狗数据集链接 提取码: gter

拿到数据后用split_data.py做分流

net.py需要和train.py在统一路径下,更改train.py里的路径直接运行train.py即可。

代码实现

在测试中猫狗的预测精度可达到90%