keras 利用vgg16进行十分类模型训练

最新推荐文章于 2024-07-31 17:37:25 发布

wwxy261

最新推荐文章于 2024-07-31 17:37:25 发布

阅读量6.2k

点赞数 1

分类专栏：算法

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

版权

导入模块

import os
import numpy as np
import tensorflow as tf
import random
import seaborn as sns
import matplotlib.pyplot as plt

from keras.models import Sequential, Model
from keras.layers import Dense, Dropout, Activation, Flatten, Input
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.utils import np_utils
from keras.applications.vgg16 import VGG16, preprocess_input
from sklearn.model_selection import train_test_split

读取图片函数

def read_and_process_image(data_dir,width=32, height=32, channels=3, preprocess=False):
    
    train_classes= [data_dir +  i for i in os.listdir(data_dir) ]
    train_images = []
    for train_class in train_classes:
        train_images=