这是一个基于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即可。