第四次作业:猫狗大战挑战赛

一、VGG模型迁移学习

开始前检测有无GPU设备:

 

1、下载数据:

 

2、数据处理:

 

 

 

 

 

 

 

 

3、创建VGGV Model:

 

 

 

 

4、修改最后一层,冻结前面层的参数:

 

5、训练并测试全连接层:

 

6、可视化模型预测结果:

 

二、上传AI研习社进行测试

 

 

posted @ 2021-10-24 21:30  小鸭鸭吖  阅读(50)  评论(0)    收藏  举报