猫狗数据读取CNN模型并进行加载

猫狗数据读取CNN模型并进行加载

import numpy as np
import pickle
import cv2
import pandas as pd
import tensorflow as tf
import matplotlib.pyplot as plt

train_epochs = 12  # 训练轮数
batch_size = 40
display_step = 10  # 显示训练结果的间隔
learning_rate = 0.00001  # 学习效率
drop_prob = 0.2  # 正则化,丢弃比例
fch_nodes = 256

train_data = {b'data': [], b'labels': []}
with open("D:/TensorFlow_gpu/animal.pickle", mode='rb') as file:
    data = pickle.load(file, encoding='bytes')
    train_data[b'data'] += list(data['train_images'])
    train_data[b'labels'] += list(data['train_label'])

x_train = np.array(train_data[b'data'])
y_train = np.array(pd.get_dummies(train_data[b'labels']))

def conv2d(x, w):
    return tf.nn.conv2d(x, w, strides=[1, 1, 1, 1], padding='SAME')
def max_pool_2x2(x):
    return tf.nn.max_pool(x, ksize=[1, 2, 2, 1], st