AchDream的博客

程序使用kaggle中的猫狗大战train data set，用来生成tfrecords数据。原数据库共有25000张照片使用原来的train数据库,然后分为train.tfrecords和test.tfrecords两个数据集。train.tfrecords包含23000张照片,test.tfrecords包含另外2000张照片代码块代码块语法遵循标准markdown代码...

读取tfrecords来输入给网络进行训练，要比直接从照片源文件中读取数据要快的多。import tensorflow as tfdef read_and_decode(filename, batch_size, capacity, min_after_dequeue): filename_queue = tf.train.string_input_producer([filename])

变量名与tensorflow官方中的VGG-Net模型变量名一致(VGG-16)所以最好使用官方预训练好的参数，不然训练时间太慢了。import tensorflow as tfimport get_dataimport numpy as npimport oslog_dir = "vgg_checkpoint/"BATCH_SIZE = 32'''４，batch_...