Python全系列 教程
3567个小节阅读:5931.5k
目录
鸿蒙应用开发
C语言快速入门
JAVA全系列 教程
面向对象的程序设计语言
Python全系列 教程
Python3.x版本,未来主流的版本
人工智能 教程
顺势而为,AI创新未来
大厂算法 教程
算法,程序员自我提升必经之路
C++ 教程
一门通用计算机编程语言
微服务 教程
目前业界流行的框架组合
web前端全系列 教程
通向WEB技术世界的钥匙
大数据全系列 教程
站在云端操控万千数据
AIGC全能工具班
A A
White Night
ximport torch
import torch.nn as nn
import torch.optim as optim
import torchvision
import matplotlib.pyplot as plt
import os
from torchvision import datasets,transforms,models
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
# 设置预处理方式
data_transform = transforms.Compose([
transforms.RandomHorizontalFlip(), # 对训练样本进行随机水平翻转
transforms.ToTensor(), # 转化成Tensor
transforms.Normalize((0.5,),(0.5,)) # 对每个通道按照指定均值和标准差进行归一化
])
# 加载数据集("井")
trainset = datasets.ImageFolder(root="faces",transform=data_transform)
# 数据加载器("桶")
trainloader = torch.utils.data.DataLoader(trainset,batch_size=5,
shuffle=True)