-
ingestion
-
crop_celeba_aligned.py
- Celaba数据集的预处理,图像裁剪,存储为h5格式等
-
dataset_tool_modify.py
-
Tensorflow的数据集构造工具
- create_mnist
- create_mnistrgb
- create_cifar10
- create_cifar100
- create_svhn
- create_lsun
- create_celeba
- create_celebahq
-
process_celeba.py
- 下载和预处理celaba,cifar10,lsun等数据集
-
process_celeba_hd.py
- 下载和抽取celaba wild数据集
-
process_transient_attribute_scenes.py
- 下载和解压transient_attribute_scenes数据
-
model
-
pggan
-
dataset.py
- 解析和构造tfrecords数据集的
-
dataset_tool.py
- tfrecords的导出和添加图片到tfrecords
-
legacy.py
- 用以加载老的Theano模型
-
loss.py
- WGAN + AC-GAN的D/G的损失函数
-
misc.py
- 辅助函数,图片/模型/结果等保存和加载。日志
-
network.py
- 论文中的网络结构,所有的卷积函数等
-
tfutil.py
- tensorflow实现的各种基础函数,数学运算类
-
train.py
- 训练网络
-
util_scripts.py
-
generate_fake_images
- 根据模型id载入模型并随机生成图像
-
generate_interpolation_video
- 训练过程中合成图像的变化视频
- generate_training_video
-
cnn_face_attr_celeba.py
- 构造MobileNet,并使用celaba数据集训练。按照celaba数据的多分类模型
-
tl_gan
-
bokeh_webgui_server.py
- 载入所有的特征方向,生成图像,画出界面
-
feature_axis.py
-
find_feature_axis
- 训练特征模型线性/非线性,返回特征权重
-
normalize_feature_axis
- 特征轴normalization化
-
disentangle_feature_axis
- 使其他特征轴与基准特征轴(性别/年龄)正交
- disentangle_feature_axis_by_idx
-
orthogonalize_one_vector
- 调整某个特征轴使得它与基准特征轴正交
-
orthogonalize_vectors
- 使所有的得特征轴两两正交
- plot_feature_correlation
- plot_feature_cos_sim
-
script_gen_sample_pggan.py
- 载入模型,随机生成批量图像并保存对应的latents
- 此处会保存 latent z和对应的图片x,一起存储为pkl
-
script_transform_sample_pickle_to_img.py
- 从pkl文件中载入图片x存储为jpg和对应的z,并存储为npy和sample_z.h5
- 为对应的图片,存储其latents为sample_z.h5
-
script_predict_labels.py
- 载入celaba属性分类模型MobieNet,预测所有样本图片x的分类标签y并存储。
- 根据jpg图像x 预测所有的标签y:sample_y.h5
存储图片到指定路径
-
script_label_regression.py
- 根据生成的样本y(特征向量)和z(对应的latent向量)找到最终的所有特征的正交轴并存储,得到feature direction
- 根据latent z和对应标签y,回归得到所有特征方向 feature_direction.pkl
-
script_generation_grid.py
- 载入模型和特征方向,在随机latent上融合方向特征,并生成对应图像并保存
-
script_generation_interactive.py
- 载入模型和特征方向文件。随机生成图像,用户可以交互地更改不同特征方向
-
script_interpolate_pggan.py
- 载入模型,生成一系列变动(特征轻微变动)的图像。
-
feature_celeba_organize.py
- 所有特征的名字 序号等 配置信息
-
generate_image.py
- 载入模型,随机生成一张图像
-
notebooks
- tl_gan_ipywidgets_gui.ipynb
- tests