文件名称:利用VGG16对kaggle比赛提供的猫狗图片进行迁移学习-PyTorch版本
文件大小:91KB
文件格式:ZIP
更新时间:2023-04-20 13:28:28
迁移学习 VGG16 Kaggle PyTorch 分类
利用VGG16对kaggle比赛提供的猫狗图片进行迁移学习-PyTorch版本 1.已改写为gpu-cpu通用 2.数据集处理参考注释;含后续训练、训练集/验证集accuracy计算,图片测试 3.数据集请参考https://www.kaggle.com/c/dogs-vs-cats
【文件预览】:
TransferLearning.py
split_train_val.py
VGG16.py
.idea
----misc.xml(304B)
----workspace.xml(12KB)
----deployment.xml(375B)
----inspectionProfiles()
--------profiles_settings.xml(174B)
----modules.xml(269B)
----VGG16.iml(451B)
__pycache__
----VGG16.cpython-37.pyc(691B)
----classes.cpython-37.pyc(271B)
predict.py
classes.py
predict_set
----12497.jpg(17KB)
----10.jpg(22KB)
----2.jpg(17KB)
----12494.jpg(27KB)