简介基于PyTorch在CelebA数据集上训练DCGAN的完整实战项目源码包面向对生成对抗网络感兴趣的初学者与希望提升工程能力的开发者。项目从理论到实践系统涵盖DCGAN网络结构搭建、CelebA数据预处理、训练流程与结果可视化等关键环节配套详细说明文档可直接运行复现。资源共12个文件以4个Python脚本为核心分别负责模型定义、训练、生成与工具封装并包含训练完成的模型权重、损失曲线及生成图像的图片与动图可视化结果整体约102.4兆字节结构清晰便于按需查阅。目前已吸引三百九十九人学习下载兼具参考与复现价值。通过本项目可掌握生成对抗网络的核心原理与实际调参方法学会监控损失变化、评估生成图像质量并能基于名人面部属性数据集独立完成人脸生成实验。训练过程中的可视化文件与最终模型文件为后续迁移到其他图像生成任务提供了可复用的基线也适合进一步改造与扩展。1. 训练 DCGAN 最怕的不是模型复杂而是反复翻车我第一次用 PyTorch 在 CelebA 数据集上训练 DCGAN 时模型结构和损失函数都照着论文抄结果跑到第 5 个 epoch生成图像还是一片模糊色块。后来拿到这份项目源码才发现问题不在网络结构而在数据预处理和优化器参数上。这份资源把 DCGAN 的完整训练流程打包好了dcgan.py 定义生成器和判别器utils.py 处理数据加载train.py 跑对抗训练还附带训练好的 model_final.pth 权重、损失曲线图和一张展示训练过程的 GIF。适合刚看完 GAN 理论、想用真实数据集跑通完整流程的初学者也适合要拿 DCGAN 做基线实验、想在自采数据集上复现图像生成的中级开发者。下面按代码结构、环境复现、训练踩坑、进阶验证这条线拆开讲每一部分都给出能直接照抄的改法和参数建议。2. 读懂 PyTorch 源码结构生成器、判别器与训练循环的三层拆解拿到压缩包先别急着训练花十分钟把几个 Python 文件的关系理清楚。这份项目遵循 DCGAN 代码的经典组织方式模型、工具、训练三个脚本各司其职理解了这个结构后面改任何参数都不会像无头苍蝇一样乱试。2.1 dcgan.py生成器的转置卷积与判别器的 LeakyReLU 设计生成器的本质是把低维随机噪声映射到高维图像空间。项目沿用了 DCGAN 论文里的推荐结构用 5 层转置卷积把 100 维噪声向量逐步放大成 64×64 的三通道图像。转置卷积并不是卷积的逆运算它通过在输入像素间补零再执行普通卷积来放大特征图尺寸参数里 kernel size、stride、padding 三者的配合直接决定每一层输出尺寸。下面这段是生成器的核心定义# dcgan.py - 生成器从 100 维噪声生成 64x64 人脸图像 class Generator(nn.Module): def __init__(self, latent_dim100, ngf64): super(Generator, self).__init__() self.main nn.Sequential( # 输入 (latent_dim, 1, 1)输出 (ngf*8, 4, 4) nn.ConvTranspose2d(latent_dim, ngf * 8, 4, 1, 0, biasFalse), nn.BatchNorm2d(ngf * 8), nn.ReLU(True), # 输出 (ngf*4, 8, 8) nn.ConvTranspose2d(ngf * 8, ngf * 4, 4, 2, 1, biasFalse), nn.BatchNorm2d(ngf * 4), nn.ReLU(True), # 输出 (ngf*2, 16, 16) nn.ConvTranspose2d(ngf * 4, ngf * 2, 4, 2, 1, biasFalse), nn.BatchNorm2d(ngf * 2), nn.ReLU(True), # 输出 (ngf, 32, 32) nn.ConvTranspose2d(ngf * 2, ngf, 4, 2, 1, biasFalse), nn.BatchNorm2d(ngf), nn.ReLU(True), # 输出 (3, 64, 64) nn.ConvTranspose2d(ngf, 3, 4, 2, 1, biasFalse), nn.Tanh() )latent_dim 是输入噪声维度项目默认取 100这是 DCGAN 原文的标准值ngf 是特征图基数默认 64显存不足时可以降到 32想提升生成质量则可以调到 128。每层之间夹着 BatchNorm2d这也是 DCGAN 训练稳定的关键——生成器如果不做 BatchNorm深层特征分布会漂移图像颜色和对比度会变得很不稳定。最后一层用 Tanh 而不是 Sigmoid因为输出需要落在 [-1, 1] 区间与归一化后的真实图像对齐。判别器是生成器的镜像结构方向相反接收 3×64×64 的图像用普通卷积逐步下采样到 1×1输出一个 0 到 1 之间的概率值。它有两个改动需要特别留意激活函数用斜率为 0.2 的 LeakyReLU 而不是 ReLU避免负值区间的梯度完全归零输入层不加 BatchNorm这是 DCGAN 论文里明确提到的细节第一层直接加 BatchNorm 会引入不必要的批次相关性影响判别器的判别能力。2.2 utils.py数据预处理与 DataLoader 配置DCGAN 对输入数据有两个硬性要求图像尺寸统一、像素值域与生成器输出对齐。CelebA 原始图片是 178×218 的彩图直接送进网络会带来两个问题——转置卷积的尺寸递增逻辑在非方图上会失衡像素范围 [0, 255] 与 Tanh 输出的 [-1, 1] 对不上。utils.py 用一组标准变换解决这两件事# utils.py - 图像变换与 DataLoader 配置 from torchvision import transforms, datasets transform transforms.Compose([ transforms.Resize(64), # 等比缩放到短边 64 transforms.CenterCrop(64), # 中心裁剪成 64x64 方形 transforms.ToTensor(), # 转为 Tensor像素缩放到 [0, 1] transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)), # 映射到 [-1, 1] ]) dataset datasets.ImageFolder(rootdata/celeba, transformtransform) dataloader torch.utils.data.DataLoader( dataset, batch_size128, # 显存 8G 以下建议改成 64 shuffleTrue, num_workers4, # Windows 上改成 0否则容易报错 pin_memoryTrue, drop_lastTrue # 丢弃最后不足一个 batch 的样本 )Normalize 的参数 (0.5, 0.5, 0.5) 配合 (0.5, 0.5, 0.5)含义是像素值先归一化到 [0, 1]减去 0.5 再除以 0.5最终落到 [-1, 1]。这里特别提醒一点如果你做过 ImageNet 预训练模型的迁移会习惯用 mean(0.485, 0.456, 0.406) 那组统计值但 DCGAN 不能用那组必须让值域对齐生成器输出 Tanh 的区间否则判别器看到的真实图像和生成图像分布会系统性错位。drop_lastTrue 值得保留。如果最后一个 batch 样本数小于 batch_sizeBatchNorm 计算均值方差时会因为样本太少而不稳定训练早期尤其容易触发异常。ImageFolder 要求数据目录按类别分子文件夹项目里常见的组织方式是 data/celeba/celeba/ 双层目录。如果 dataset 加载出来长度是 0先检查是不是目录层级多套了一层。2.3 train.py对抗训练循环中的梯度流控制与优化器选择train.py 是整个项目的核心实现的是 GAN 训练里最经典的交替更新策略每个 iteration 先固定生成器更新判别器再固定判别器更新生成器。判别器的目标是分辨真实图像和生成图像生成器的目标则是让判别器把假图判为真。项目用 BCELoss 配合 Sigmoid 输出这是 DCGAN 原版的标准组合# train.py - 对抗训练主循环核心代码 criterion nn.BCELoss() optimizer_G torch.optim.Adam( generator.parameters(), lr0.0002, betas(0.5, 0.999) ) optimizer_D torch.optim.Adam( discriminator.parameters(), lr0.0002, betas(0.5, 0.999) ) for epoch in range(num_epochs): for i, (imgs,) in enumerate(dataloader): real_imgs imgs.to(device) real_label torch.ones(imgs.size(0), 1, devicedevice) fake_label torch.zeros(imgs.size(0), 1, devicedevice) # 第一步更新判别器 noise torch.randn(imgs.size(0), 100, devicedevice) fake_imgs generator(noise) # 生成器前向 real_pred discriminator(real_imgs) # 真实图像预测 fake_pred discriminator(fake_imgs.detach()) # 生成图像预测 loss_D criterion(real_pred, real_label) criterion(fake_pred, fake_label) optimizer_D.zero_grad() loss_D.backward() optimizer_D.step() # 第二步更新生成器 fake_pred discriminator(fake_imgs) # 重新前向不 detach loss_G criterion(fake_pred, real_label) # 目标是让判别器判真 optimizer_G.zero_grad() loss_G.backward() optimizer_G.step() if i % 100 0: print(fEpoch [{epoch}/{num_epochs}] Batch [{i}] fD_loss: {loss_D.item():.4f} G_loss: {loss_G.item():.4f})这里的梯度流控制是理解 GAN 训练的关键。更新判别器时用 fake_imgs.detach() 切断生成器计算图反向传播只更新判别器参数更新生成器时重新前向计算 fake_pred让梯度能穿过生成器的所有参数。如果忘记 detach判别器更新时会把梯度连带传回生成器两个网络的参数会在同一步内被重复更新训练必然震荡。lr0.0002 和 betas(0.5, 0.999) 是 DCGAN 论文明确指定的配置尤其 beta10.5 不能随便改。PyTorch 里 Adam 默认的 beta1 是 0.9代表对历史梯度的一阶矩估计惯性大适合普通监督学习但 GAN 的梯度方向本身不稳定惯性太大会让优化器沿着过时方向继续走训练曲线剧烈振荡。学习率可以微调但一般不超过 0.0002低于 0.0001 则收敛变慢。3. 从零复现训练环境PyTorch 安装、CelebA 数据准备与参数调优要把这份源码跑起来需要解决三个现实问题PyTorch 版本怎么选不翻车、CelebA 数据目录怎么组织、训练参数调到什么程度合适。这一章按顺序给出可直接执行的命令和判断标准。3.1 环境搭建PyTorch、CUDA 与显卡驱动的版本匹配PyTorch 安装的坑多半不在 pip 命令本身而在版本匹配。我一般的检查顺序是先看显卡驱动支持的最高 CUDA 版本再选对应编译版本的 PyTorch。30 系及以上的显卡驱动通常能支持 CUDA 11.8 或 12.x装官方源里的 cu118 或 cu121 版本都没问题10 系老卡建议用 cu118 以下的版本。# 查看显卡驱动信息和 CUDA 版本 nvidia-smi # 右上角 CUDA Version 是驱动支持的最高版本不是已安装的 CUDA # 创建虚拟环境并安装 PyTorch以 CUDA 11.8 为例 conda create -n dcgan python3.9 -y conda activate dcgan pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118这里有一个高频误区nvidia-smi 显示的 CUDA 版本只是驱动支持的上限不代表你要额外安装对应版本的 CUDA toolkit。PyTorch 在 pip 安装时自带 CUDA 运行库只要驱动版本不低于 PyTorch 的要求就能跑。验证 PyTorch 是否真正用上 GPU用下面这段代码不要只看安装日志# 验证 PyTorch GPU 可用性 import torch print(torch.__version__) print(torch.cuda.is_available()) # 必须为 True print(torch.cuda.get_device_name(0)) # 显示实际显卡型号 device torch.device(cuda if torch.cuda.is_available() else cpu)如果 torch.cuda.is_available() 返回 False常见原因有三个装成了 CPU 版 PyTorchpip install torch 默认 CPU 版、驱动版本太旧、或者 conda 环境里存在 CPU 版和 GPU 版混乱。Windows 上最简单可靠的路径是直接装官方预编译的 GPU wheel 包不要先手动装 CUDA toolkit 再装 PyTorch那样反而容易版本冲突。CPU 版也能跑通训练但 CelebA 全量有几万张图CPU 一个 epoch 可能要几十分钟建议至少用一张 4G 显存以上的显卡。3.2 CelebA 数据集下载与目录组织CelebA 数据集包含 20 多万张名人面部图片每张图带 40 个属性标注数据量大、质量高、标注完整是 GAN 图像生成最常用的真实人脸数据集之一。项目用的是全量数据训练第一次复现建议先用 5000 到 10000 张子集跑通流程再上全量。下载解压后目录要按 ImageFolder 要求的格式组织# 数据目录结构ImageFolder 要求 data/ └── celeba/ ├── img_align_celeba/ # 放全部图片 │ ├── 000001.jpg │ ├── 000002.jpg │ └── ... └── list_attr_celeba.txt # 属性标注本训练不用但保留由于 CelebA 里的图片都是正脸人脸不需要按类别分子目录数据集 root 指向外层 data/celeba 即可。ImageFolder 会自动遍历所有子目录下的图片。加载不到图的常见原因就是路径写错比如把 root 写成了 data/celeba/img_align_celeba 而不是 data/celeba或者下载的压缩包解压后多了一层同名目录。还有一个细节CelebA 的文件名不是严格从 000001 连续编号的不要在代码里假设索引和文件名一一对应。3.3 训练参数从默认配置到自定义修改项目默认参数基本是 DCGAN 原文的标准值直接跑可以复现论文效果但完全照抄在个人显卡上会碰显存瓶颈。参数调整优先动三个batch_size、num_epochs、latent_dim。batch_size默认 1288G 显存建议降到 644G 显存建议 32。改小 batch 后学习率要同步微调因为小 batch 的梯度估计噪声更大建议 lr 从 0.0002 降到 0.0001。num_epochsCelebA 全量数据一般 20 个 epoch 左右能看到清晰人脸。前 5 个 epoch 出人脸轮廓10 个 epoch 后五官逐渐清晰。验证代码是否跑通用 3 个 epoch 加几千张子集图就够了。latent_dim100 维是标准值不用改。想增大潜在空间容量可以加到 128但更高维度会让生成器更难收敛收益不一定明显。如果改了图像尺寸、网络结构或者 latent_dim项目附带的 model_final.pth 权重就不能直接加载了需要重新训练。模型权重和代码配置是严格绑定的这一点在后续做迁移实验时要记住。4. 训练避坑指南损失异常、模式塌缩与显存不足的排查记录这一章写的都是我复现过程中真实踩过的坑每一条按「现象 → 原因 → 解决」展开可以对照自己的训练日志逐项排查。这些坑单独看都很小但任何一个都能让训练白跑几个晚上。4.1 现象判别器损失直接掉到 0生成器梯度消失训练刚开始几百个 iterationD_loss 降到 0.001 以下G_loss 一直飘在 10 以上生成图像全是噪点没有任何结构。原因是判别器太强真实图像和生成图像的分布被完全分离开生成器无论输出什么都被一眼识破梯度失去了指导意义。这是 GAN 训练里最经典的均衡问题——生成器和判别器的能力必须此消彼长地同步提升。解决的办法按顺序试三个。先把生成器的学习率单独调到 0.0004缩小和判别器的能力差距不行就给判别器加 Dropout一般 0.3 到 0.5降低它对训练集的记忆能力最后可以改成每训练两次生成器再训练一次判别器给生成器更多学习机会。项目源码里没有默认给判别器加 Dropout自己加的时候放在每个卷积块之后、激活之前。4.2 现象生成图像全部塌缩成同一种脸型训练到第 10 个 epoch生成出来的图每张都是几乎相同的脸姿态、肤色、表情都差不多这就是 Mode Collapse。原因是生成器找到了判别器的盲区——只要输出一种能稳定骗过判别器的图像就停留在原地不再探索其他模式。CelebA 这种类别单一但姿态多样的人脸数据尤其容易出现因为判别器对全局结构的判断比局部细节更敏感。处理 Mode Collapse 没有一击必杀的方案常见做法是组合拳给 latent 输入加噪声把噪声向量一部分元素随机置零或者调整训练比例让判别器更严格。更实用的做法是在每个 epoch 末尾保存一批生成图像像项目里 Generated_Epoch_1.png 到 Generated_Epoch_10.png 那样连续对比如果从第 5 个 epoch 开始面孔就不再变化基本可以确认塌缩。重新训练时把 z 的维度加大或者把判别器加深增加它的判别粒度都有一定帮助。4.3 现象训练中途显存溢出进程直接崩溃报错信息是 CUDA out of memory往往发生在某个 epoch 的中段。显存不足最直接的原因是 batch_size128 加 64×64 图像尺寸中间特征图占用超出显卡容量。尤其判别器要同时处理真实图像和生成图像的梯度显存占用接近双倍。# 第一步确认是训练占用还是缓存占用 nvidia-smi # 看进程列表里 Python 进程的显存占用如果高于 80% 基本就是训练本身超了解决方式按效果排序batch_size 降到 64 或 32同时把 DataLoader 的 pin_memory 改成 False 减少额外锁页内存还不够就把 num_workers 降到 2 或 0仍然溢出只能把图像输出尺寸从 64 降到 48但需要同步修改生成器和判别器的卷积层参数新手不建议折腾。PyTorch 2.x 支持混合精度训练但 DCGAN 对精度变化敏感我在训练里不主动开 AMP容易诱发 NaN。4.4 现象图像训练到一半不再变化loss 卡在某个固定值生成图像停留在半清晰状态D_loss 和 G_loss 来回拉锯但图像没有本质提升。这种停滞通常是两个原因叠加一是学习率太小后期优化器步长不足以逃离局部平衡二是 BatchNorm 的 running_mean 和 running_var 在训练后期变得非常稳定生成器对输入的响应趋于固化。我的做法是在训练中段把 lr 手动降一个数量级到 0.00002让优化器进入精细化调整阶段这个操作对恢复图像细节效果明显。想要更自动的方案可以给两个优化器挂 torch.optim.lr_scheduler.CosineAnnealingLR周期设成剩余 epoch 数。注意改 lr 的时刻很关键太早会让模型停在粗糙轮廓太晚则浪费计算资源一般选在 60% epoch 处。4.5 现象loss 出现 NaN训练直接中止NaN 出现的直接原因是梯度值或损失值溢出。常见诱因有三个某个 batch 的输入数据包含异常值图像文件损坏、像素越界、网络权重初始化范围不合适、学习率过大导致激活值爆炸。CelebA 原图质量整体高但偶尔会有损坏的图片文件ImageFolder 读取时不会报错数据值却可能是错的。# 在训练循环里加一道数据检查发现非法值直接跳过 if not torch.isfinite(imgs).all(): print(fBatch {i} contains invalid values, skipped) continue处理 NaN 的第二个动作是检查权重初始化。DCGAN 原版推荐用均值为 0、标准差 0.02 的正态分布初始化卷积层和 BatchNorm 层。如果源码用的是 PyTorch 默认初始化训练到较深的层数时有概率触发梯度爆炸。补上权重初始化逻辑后NaN 问题基本能根除。5. 模型验证与进阶技巧权重加载、潜在空间插值与训练监控习惯项目附带训练好的 model_final.pth 权重路径在 model 目录下直接加载就能生成 CelebA 风格人脸不需要重新训练。这部分讲两个不改变网络结构就能做的验证实验以及一个我坚持了很久的训练监控习惯。5.1 用 training 好的权重批量生成新图像项目的 generate.py 就是干这件事的加载权重、生成随机噪声、前向推理、保存图像。下面是兼容 CPU 加载 GPU 权重的方式# generate.py - 加载训练好的模型并生成新图像 import torch from torchvision.utils import save_image from dcgan import Generator latent_dim 100 device cuda if torch.cuda.is_available() else cpu generator Generator(latent_dim).to(device) # 训练时在 GPU 上保存的权重CPU 加载必须指定 map_location generator.load_state_dict(torch.load(model/model_final.pth, map_locationdevice)) generator.eval() with torch.no_grad(): noise torch.randn(64, latent_dim, devicedevice) fake_imgs generator(noise) # 输出值域是 [-1, 1]需映射回 [0, 1] 才能正常显示 save_image((fake_imgs 1) / 2, generated_new.png, nrow8)两个细节值得记住。第一map_location 参数必须加否则在 CPU 机器或不同型号 GPU 上会报 device 不匹配错误。第二生成器输出值域是 [-1, 1]直接保存会得到偏暗的图像先加 1 再除以 2 恢复日常的像素范围。eval() 模式会固定 BatchNorm 的统计量保证多次生成结果稳定。5.2 潜在空间插值实验检验特征空间是否连续这是一个能直观检验生成器质量的实验取两个随机噪声向量 z1 和 z2在它们之间线性插值每一步生成一张图像连续看时脸型、表情应该平滑过渡。如果中间突然跳变或产生扭曲面孔说明特征空间断裂模型训练不充分。# 潜在空间插值演示 z1 torch.randn(1, latent_dim, devicedevice) z2 torch.randn(1, latent_dim, devicedevice) for t in range(11): # 在 z1 和 z2 之间取 11 个点 z_interp z1 (z2 - z1) * (t / 10) with torch.no_grad(): img generator(z_interp) save_image((img 1) / 2, finterp_{t:02d}.png)插值平滑说明生成器学到了连续且语义化的人脸特征空间比如表情、姿态、光照在插值中会渐进变化如果图像在某个点突然崩坏通常意味着训练 epoch 不够或者发生了模式塌缩。这个实验比单看几张生成图更能暴露模型的真实状态。最后一个建议也算是我踩过无数次坑之后形成的习惯每次训练开始前固定一组噪声种子和采样间隔每隔固定 iteration 就用这些固定噪声生成一版图像类似项目里 Generated_Epoch_1.png 到 Generated_Epoch_10.png 的做法。这样能把训练过程的中间状态完整串联起来翻车时翻回去看中间帧一眼就能定位模型是从哪个 epoch 开始崩的。从那以后我每跑一次 DCGAN 都会强制走一遍这个流程损失曲线和生成图结合着看模型好坏基本不再靠猜。希望这篇拆解对你跑通这份项目源码有帮助不管是新手入门还是拿去做基线实验都值得动手复现一次。本文还有配套的精品资源点击获取