VGG神经网络及pytorch初下实测

# -*- coding: utf-8 -*-
"""
Created on Mon May  9 19:16:47 2022

@author: liaoy
"""

import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers, regularizers
import numpy as np
import os
import cv2
import matplotlib.pyplot as plt


os.environ["CUDA_VISIBLE_DEVICES"] = "1"
resize = 224
path ="D:\\spyder77\\train"

def load_data():
    imgs = os.listdir(path)
    num = len(imgs)
    train_data = np.empty((500, resize, resize, 3), dtype="int32")
    train_label = np.empty((500, ), dtype="int32")
    test_data = np.empty((500, resize, resize, 3), dtype="int32")
    test_label = np.empty((500, ), dtype="int32")
    for i in range(500):
        if i % 2:
            train_data[i] = cv2.resize(cv2.imread(path+'/'+ 'dog.' + str(i) + '.jpg'), (resize, resize))
            train_label[i] = 1
        else:
            train_data[i] = cv2.resize(cv2.imread(path+'/' + 'cat.' + str(i) + '.jpg'), (resize, resize))
            train_label[i] = 0
    for i in range(500, 1000):
        if i % 2:
            test_data[i-500] = cv2.resize(cv2.imread(path+'/' + 'dog.' + str(i) + '.jpg'), (resize, resize))
            test_label[i-500] = 1
        else:
            test_data[i-500] = cv2.resize(cv2.imread(path+'/' + 'cat.' + str(i) + '.jpg'), (resize, resize))
            test_label[i-500] = 0
    return train_data, train_label, test_data, test_label
def vgg16():
    weight_decay = 0.0005
    nb_epoch = 100
    batch_size = 32
    
    # layer1
    model = keras.Sequential()
    model.add(layers.Conv2D(64, (3, 3), padding=