数据集:
- 共2000张图片,1000张狗,1000张猫;
- 1400张用于训练,600张用于测试;
- 图片均为 RGB 3 通道,尺寸大小不一。
数据链接:https://pan.baidu.com/s/15E9D6qztrY3bfZp5kil2Fw
提取码:8qin模型:
- 卷积层:卷积核大小3*3,个数64个;
- 卷积层:卷积核大小3*3,个数64个;
- 卷积层:卷积核大小3*3,个数128个;
- 全连接层
- 全连接层(输出层)
'''
train.py
'''
import dataset
import tensorflow as tf
import numpy as np
import time
import math
import random
batch_size = 32 # 每批样本数
validation_size = 0.3 # 测试集所占比例
img_size = 64 # 图片尺寸大小
num_channels = 3 # 图像通道数(RGB)
train_path &