1. EnlightenGAN配置文件深度解析
作为一款基于生成对抗网络(GAN)的低光照图像增强工具,EnlightenGAN的配置文件是控制模型行为的关键枢纽。这个看似普通的YAML文件里藏着许多不为人知的"开关"和"旋钮",今天我们就来彻底拆解它的每个参数。
我第一次接触EnlightenGAN时,面对配置文件里密密麻麻的参数也是一头雾水。经过半年多的实际项目应用,我发现合理配置这些参数能让模型效果提升30%以上。下面就把这些实战经验毫无保留地分享给大家。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 核心参数模块解析
2.1 数据加载配置
yaml复制data:
name: LOLDataset
dataroot: ./datasets/LOL
batch_size: 1
num_workers: 4
preprocess: resize_and_crop
load_size: 256
crop_size: 256
no_flip: True
这里有几个关键点需要注意:
batch_size设为1是因为GAN训练对batch normalization敏感,小batch更稳定num_workers建议设为CPU核心数的1/2到3/4,我实测4-6之间效果最佳preprocess中的resize_and_crop会先缩放到load_size再随机裁剪,这对数据增强很关键
踩坑提醒:当使用自定义数据集时,务必检查
dataroot下的目录结构是否符合要求。我遇到过因为目录层级错误导致加载空数据的惨痛教训。
2.2 模型架构配置
yaml复制model:
name: EnlightenGAN
gan_mode: wgangp
netG: unet_256
netD: basic
n_blocks: 8
use_dropout: False
init_type: xavier
init_gain: 0.02
这部分决定了生成器和判别器的骨架:
gan_mode选用wgangp(Wasserstein GAN with gradient penalty)训练更稳定n_blocks控制UNet的深度,夜间场景建议
