深度学习------戴口罩和不戴口罩

本博客主要通过vgg16,对戴口罩数据集进行二分类；
步骤：1.加载数据集，对数据集进行预处理
2.加载模型
3.对模型进行训练

数据集参考百度网盘：
链接：https://pan.baidu.com/s/1dvv64GxwlMN7u1rBgWeT5A
提取码：j4xi

代码如下：

import random
import torch
import torch.nn as nn
import torchvision.transforms as transform
from torch.utils import