基于TensorFlow的Cats vs. Dogs(猫狗大战)实现和详解(2)
2. 卷积神经网络模型的构造——model.py 关于神经网络模型不想说太多,视频中使用的模型是仿照TensorFlow的官方例程cifar-10的网络结构来写的。就是两个卷积层(每个卷积层后加一个池化层),两个全连接层,最后一个softmax输出分类结果。import tensorflow as tfdef inference(images, batch_size, n_cla...