当前位置：网站首页>深度学习------戴口罩和不戴口罩

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

2022-08-04 11:21:00 【小飞龙程序员】

本博客主要通过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
原网站

版权声明
本文为[小飞龙程序员]所创，转载请带上原文链接，感谢
https://blog.csdn.net/m0_47405013/article/details/125869148

边栏推荐

猜你喜欢

随机推荐