![](https://img-blog.csdnimg.cn/20201014180756925.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tensorflow
AchDream
这个作者很懒,什么都没留下…
展开
-
tensorflow使用猫狗大战数据库生成tfrecords数据
程序使用kaggle中的猫狗大战train data set,用来生成tfrecords数据。原数据库共有25000张照片使用原来的train数据库,然后分为train.tfrecords和test.tfrecords两个数据集。train.tfrecords包含23000张照片,test.tfrecords包含另外2000张照片代码块代码块语法遵循标准markdown代码...原创 2018-04-22 16:19:03 · 834 阅读 · 1 评论 -
tfrecords数据读取
读取tfrecords来输入给网络进行训练,要比直接从照片源文件中读取数据要快的多。import tensorflow as tfdef read_and_decode(filename, batch_size, capacity, min_after_dequeue): filename_queue = tf.train.string_input_producer([filename])原创 2018-04-22 16:25:44 · 340 阅读 · 0 评论 -
Tensorflow实现VGG-16
变量名与tensorflow官方中的VGG-Net模型变量名一致(VGG-16)所以最好使用官方预训练好的参数,不然训练时间太慢了。import tensorflow as tfimport get_dataimport numpy as npimport oslog_dir = "vgg_checkpoint/"BATCH_SIZE = 32'''4,batch_...原创 2018-04-22 16:30:09 · 3608 阅读 · 2 评论