数据集:training_set, test-set,traing_set里面存放的是猫狗分类的数据集,test_set里面存放的是猫狗分类的测试集。
datasets.py:数据集的读取,并且会按照7:3的比例将traing_set划分为测试集和验证集
chuli.py:验证数据集读取的正确性
model.py:里面存放的ResNet34的代码
train.py:训练集,并且会drew出训练集和验证集的损失和acc
test.py:最终会输出训练好的模型(resnet.pth)对于测试集的acc
该项目非常适合初学深度学习者,可以学习关于数据集
1