python第51天

1.读取数据

使用CIFAR-10图像数据

python 复制代码
import torch
from torchvision import datasets, transforms

# 定义图像预处理流程
image_transform = transforms.Compose([
    transforms.ToTensor(),  # 将PIL图像转换为张量
    transforms.Normalize(mean=(0.5, 0.5, 0.5),  # RGB三通道均值
                         std=(0.5, 0.5, 0.5))   # RGB三通道标准差
])

# 获取训练数据集
trainset = datasets.CIFAR10(
    root='./data',  # 数据集存储路径
    train=True,     # 使用训练集
    transform=image_transform,
    download=True   # 如果本地不存在则下载
)

# 获取测试数据集
testset = datasets.CIFAR10(
    root='./data',
    train=False,    # 使用测试集
    transform=image_transform,
    download=True
)

# 配置数据加载器
train_loader = torch.utils.data.DataLoader(
    dataset=trainset,
    batch_size=128,    # 每批样本数量
    shuffle=True       # 训练时打乱顺序
)

test_loader = torch.utils.data.DataLoader(
    dataset=testset,
    batch_size=128,
    shuffle=False      # 测试时保持原始顺序
)

2.模型建立

(1)建立CNN模型

python 复制代码
import torch
import torch.nn as nn
 
class SimpleCNN(nn.Module):
    def __init__(self):
        super(SimpleCNN, self).__init__()
        self.conv1 = nn.Conv2d(3, 16, 3, padding=1)
        self.pool = nn.MaxPool2d(2, 2)
        self.conv2 = nn.Conv2d(16, 32, 3, padding=1)
        self.fc1 = nn.Linear(32 * 8 * 8, 256)
        self.fc2 = nn.Linear(256, 10)
        self.relu = nn.ReLU()
 
    def forward(self, x):
        x = self.pool(self.relu(self.conv1(x)))  # 16x16x16
        x = self.pool(self.relu(self.conv2(x)))  # 32x8x8
        x = x.view(-1, 32 * 8 * 8)
        x = self.relu(self.fc1(x))
        x = self.fc2(x)
        return x
 

@浙大疏锦行

相关推荐
CHANG_THE_WORLD24 分钟前
「macOS 系统字体收集器 (C++17 实现)」
开发语言·c++·macos
妄想出头的工业炼药师43 分钟前
python和C++相互调用使用
开发语言·c++
2301_764441331 小时前
Python管理咨询数据可视化实战:收入分布与顾问利用率双轴对比图表生成脚本
开发语言·python·信息可视化
景彡先生1 小时前
C++17 并行算法:std::execution::par
开发语言·c++
该用户已不存在1 小时前
不知道这些工具,难怪的你的Python开发那么慢丨Python 开发必备的6大工具
前端·后端·python
nextera-void1 小时前
仓颉编程语言:从入门到精通
开发语言
神经星星1 小时前
AI 论文周报 | Chai-2刷新抗体设计效率,命中率提高100倍;多篇ICML入围论文一键速览
人工智能·深度学习·机器学习
oioihoii1 小时前
C++随机打乱函数:简化源码与原理深度剖析
开发语言·c++·算法
水果里面有苹果2 小时前
19-C#静态方法与静态类
java·开发语言·c#
Monkey的自我迭代2 小时前
Python标准库:时间与随机数全解析
前端·python·数据挖掘