GitHub - pingxi1009/ResNet50: ResNet50 猫狗数据集训练

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
Model.pth		Model.pth
README		README
ResNet50.py		ResNet50.py
dataload.py		dataload.py
loss.png		loss.png
test.txt		test.txt
train.py		train.py
正确率.png		正确率.png

Repository files navigation

文件说明：
1、ResNet50.py 是构建模型 ResNet50
2、dataload.py 是用来加载数据集
3、train.py 是训练模型
4、Model.pth 是训练好的模型，可直接加载使用
5、两张 PNG 图片为记录一次训练的成果

使用须知：
1、我的环境是 (win10 64位) + (Python 3.8.3) + (OpenCV 4.5.1) + (Pytorch 1.7.0+cu110)
2、要使用的话直接运行 train.py 即可
3、欢迎沟通交流，一起进步，共同成长1009088103@qq.com