keras 猫狗大战自搭网络以及vgg16应用

最新推荐文章于 2024-09-29 20:15:10 发布

wwxy261

最新推荐文章于 2024-09-29 20:15:10 发布

阅读量1.3k

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/wwxy1995/article/details/81455688

版权

导入模块

import os
import numpy as np
import tensorflow as tf
import random
import seaborn as sns
import matplotlib.pyplot as plt
import keras
from keras.models import Sequential, Model
from keras.layers import Dense, Dropout, Activation, Flatten, Input,BatchNormalization
from keras.layers.convolutional import Conv2D, MaxPooling2D
from keras.optimizers import RMSprop, Adam, SGD
from keras.preprocessing import image
from keras.preprocessing.image import ImageDataGenerator
from keras.applications.vgg16 import VGG16, preprocess_input

from sklearn.model_selection import train_test_split

加载数据集

def read_and_process_image(data_dir,width=64, height=64, channels=3, preprocess=False):
    train_images= [data_dir + i for i in os.listdir(data_dir)]
    
    random.shuffle(train_images)
    
    def read_image(file_path, preprocess):
        img