1. 首页
  2. 人工智能
  3. 深度学习
  4. VGG16 实现 图像分类 tensorflow python

VGG16 实现 图像分类 tensorflow python

上传者: 2019-04-05 04:30:24上传 PY文件 10.24KB 热度 98次
这是利用tensorflow实现的,利用卷积神经网络 VGG 进行图像分类的简单程序,分享给大家供交流学习使用,如果代码有什么问题和错误,欢迎大家多提宝贵意见,详情请参考本人博客 适用数据集也进行了上传,在本人资源页可以找到 这个程序相对简单易用,自己的数据仅需按照不同类别(1,2,...,n)放置在如下格式目录即可: 训练数据目录 ./example/train/1 , ./example/train/2 ,... , ./example/train/n 测试数据目录 ./example/test/1 , ./example/test/2 ,... , ./example/test/n 更简单的,如果只有一个数据集,通过交叉验证的方式训练并测试 目 录  ./easy/1 , ./easy/2 ,... , ./easy/n 代码中,类别书,图像格式,训练epoch总数,收敛判断条件,以及网络超参数需要大家根据自己的数据自行调整试验,注意如果交叉验证的话(对应代码中one dataset cross validation ),屏蔽掉训练集和测试集部分代码(对应代码中train and test set),反之亦然 详情请参考本人博客 录  ./easy/1 , ./easy/2 ,... , ./easy/n 代码中,类别书,图像格式,训练epoch总数,收敛判断条件,以及网络超参数需要大家根据自己的数据自行调整试验,注意如果交叉验证的话(对应代码中one dataset cross validation ),屏蔽掉训练集和测试集部分代码(对应代码中train and test set),反之亦然 详情请参考本人博客
用户评论