utils模块¶

更新时间：2024年4月13日

简介¶

utils模块保存了“足以完成一个任务”的模块级代码。

dataset_and_dataloader.py¶

准备dataset和dataloader的模块。

模块级输入：类folder、BATCH_SIZE、NUM_TO_LEARN1、NUM_TO_LEARN2、MODE

模块级输出：变量（类）dataloader

loss.py¶

存储计算单对图片loss、多对图片loss、criterion的模块。具体参考VAE模型中对loss的介绍。

模块级输入：函数custom_ssim, custom_kl

模块级输出：各类img2img损失函数，各类batch2batch判据

path_config.py¶

定义相对路径的模块。

模块级输出：类folder

VAE.py¶

保存VAE核心结构的模块。

模块级输出：类VAE