mige2008 / flowers5_train Goto Github PK
View Code? Open in Web Editor NEWThis project forked from fanchenxinok/flowers5_train
搭建tensorflow卷积网络训练flowers5分类模型,其中数据集是自己制作的tfrecords数据集,由于机器实在不给力只能搭建简单的卷积网络(2conv+2fullconnect)进行训练,整个网络搭建流程自己学习到的知识有:tfrecords数据集制作, 数据预处理,数据增广,训练模型数据的保存,tensorboard的使用, tfdbg的使用。数据增广方面除了使用tensorflow提供的函数,自己用tf.py_func定义了一个图片旋转的op, 可以随机旋转-20到20度。