手把手操作VGG16检测新冠肺炎（附完整代码和数据集）

手把手操作VGG16检测新冠肺炎（附完整代码和数据集）

前言

很久没有打代码，今天去kaggle下载了新冠肺炎数据集来练练手。顺便记录一下所学内容。
先介绍一下我的环境是：
tensofflow2.4-gpu
python3.6
1050ti
Jupyter

1-头文件

import  tensorflow as tf
from    tensorflow import  keras
from    tensorflow.keras import datasets, layers, optimizers, models
from    tensorflow.keras import regularizers
import  os
import  argparse
import  numpy as np
import  random
import  pathlib

2-VGG16模型

先上网上找到的VGG16模型图

class VGG16(models.Model):


    def __init__(self, input_shape):
        """
		对数据集的大小在后面统一resize成了(150, 150, 3)
        :param input_shape: [150, 150, 3]
        """
        super(VGG16, self).__init__()

        weight_decay = 0.000
        self.num_classes = 2  # 这里是你的分类数量，我是二分类

        model = models.Sequential()

        model.add(layers.Conv2D(64, (3, 3), padding='same',
                         input_shape=input_shape, kernel_regularizer=regularizers.l2(weight_decay)))
        model.add(layers.Activation('relu'))


        model.add(layers.Conv2D(64, (3, 3), padding='same',kernel_regularizer=regularizers.l2(weight_decay)))
        model.add(layers.Activation('relu'))
        model.add(layers.MaxPooling2D(pool_size=(2, 2)))

        model.add(layers.Conv2D(128, (3, 3), padding='same',kernel