第四次作业 CNN实战

代码练习

1.判断是否存在CPU设备

2.下载数据(训练集包含1800张图(猫的图片900张,狗的图片900张),测试集包含2000张图。)

3.将图片整理成224x224x3的大小,并进行归一化处理 同时查看dets的一些属性

4.保存并显示图片

只展示第一个batch的5张图片


5.创建 VGG Model

6.对输入的5个图片利用VGG模型进行预测,同时,使用softmax对结果进行处理


7.修改最后一层,冻结前面层的参数


8.训练并测试全连接层

训练


测试



9.可视化模型预测结果(主观分析)

posted @ 2021-10-22 15:42  张TCP  阅读(10)  评论(0)    收藏  举报