VGG+tensorflow深度卷积神经网络模型-少量图片集测试demo

java_leaf

于 2024-03-19 16:03:05 发布

阅读量1.6k

点赞数 41

文章标签： tensorflow cnn 人工智能

本文链接：https://blog.youkuaiyun.com/java_leaf/article/details/136845483

版权

图片训练集下载地址

图片资源不大，学习使用。

https://download.youkuaiyun.com/download/java_leaf/88974388编辑https://download.youkuaiyun.com/download/java_leaf/88974388

可以先看这篇文章图片训练。https://blog.youkuaiyun.com/java_leaf/article/details/136758861

VGG（Visual Geometry Group）是一种深度卷积神经网络模型

看下面的训练，少量图片在第10轮基本上准确率达到 100%，ho dear! 太惊人了。

查看VGG网络参数

vgg = VGG16(weights='imagenet',
            include_top=False, input_shape=(200, 200, 3))
vgg.summary();

Model: "vgg16"
_________________________________________________________________
 Layer (type)                Output Shape              Param #   
=================================================================
 input_1 (InputLayer)        [(None, 200, 200, 3)]     0         
                                                                 
 block1_conv1 (Conv2D)       (None, 200, 200, 64)      1792      
                                                                 
 block1_conv2 (Conv2D)       (None, 200, 200, 64)      36928