tensorflow2 inceptionv3训练猫狗数据

这里用迁移的inceptionv3网络来实现猫狗数据的训练
具体的inceptionv3网络结构可以在我的上一篇博客中找到
从kaggle上面下载数据，用其他数据也行，准确度可达99%左右，好评给个star
数据放置结构按我的GitHub上面的就行，地址在这里
导入相应的包

from tensorflow.keras.models import *
from tensorflow.keras.layers import *
from tensorflow.keras.applications import *
from tensorflow.keras.preprocessing.image import *
from tensorflow import keras as keras
import numpy as np
import cv2
from  PIL import Image
import matplotlib.pyplot as plt

导入inceptionv3模型，重新添加一个最后的全连接层，并进行训练

base_model = InceptionV3(input_tensor=Input(shape=(299, 299, 3)),weights='imagenet', include_top=False)
    #base_model.summary()
    x = base_model.output
    x = GlobalAveragePooling2D()(x)
    # let's add a fully-connected layer
    x = Dense(1024, activation='relu')(x)
    #