📸 JAX 图像生成
30章 · 从基础到项目
⚡ 友好 · 明亮色系
01
JAX基础与环境搭建
JAX简介
安装配置
vs NumPy
jit/vmap/grad
02
JAX张量操作
创建/属性
数据类型
索引切片
广播/重塑
03
自动微分与梯度计算
grad
高阶梯度
jacfwd/jacrev
vmap/pmap
04
JIT编译与性能优化
jit装饰器
静态参数
三模式
基准测试
05
随机数生成与状态管理
PRNGKey
split/fold
随机分布
vs NumPy随机
06
图像基础与表示
HWC/CHW
颜色空间
像素值范围
加载/保存
07
图像滤波与卷积
手动卷积
jax.lax.conv
高斯滤波
Sobel/Canny
08
图像变换与增强
缩放/旋转
仿射变换
色彩抖动
直方图均衡
09
图像去噪与修复
均值/中值滤波
双边滤波
非局部均值
修复基础
10
图像分割基础
阈值分割
Otsu
区域生长
分水岭
11
特征提取与描述
SIFT/HOG
LBP
JAX特征提取
特征匹配
12
图像金字塔与多尺度
高斯金字塔
拉普拉斯金字塔
多尺度融合
图像拼接
13
生成模型概述
VAE/GAN/Flow/Diffusion
JAX生态库
14
变分自编码器 (VAE)
重参数化
ELBO
Flax实现
15
生成对抗网络 (GAN)
生成器/判别器
WGAN-GP
JAX实现
16
扩散模型基础
前向/反向过程
DDPM
噪声调度
17
基于分数的生成模型
得分匹配
郎之万
NCSN
SDE
18
条件生成与引导
类别/文本条件
分类器引导
无分类器引导
19
图像超分辨率
SRCNN/ESPCN
SRGAN
扩散超分
JAX
20
图像风格迁移
神经风格迁移
AdaIN
CycleGAN
对比学习
21
图像修复与补全
部分/门控卷积
扩散修复
拉普拉斯金字塔
22
图像翻译与编辑
Pix2Pix/CycleGAN
StarGAN
扩散编辑
23
自监督学习与表示
SimCLR/MoCo
MAE
生成应用
24
Transformer在图像生成
ViT/DETR
Image Transformer
MaskGIT
25
NeRF与3D生成
NeRF原理
体渲染
Instant NGP
3D生成
26
高效生成与模型压缩
知识蒸馏
量化/剪枝
蒸馏扩散
27
JAX生态与工具链
Flax/Haiku
Optax/T5X
Orbax/XManager
28
分布式训练与TPU
pmap/shmem
数据/模型并行
TPU集群
29
模型部署与推理优化
JAX Serving
TFLite/ONNX
量化推理
30
综合项目：图像生成系统
需求分析
模型选型
训练流水线
评估部署