1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
| conf
train.yaml
test.yaml
data
__init__.py //最外层 get_dataset（dataset_name)
datasets.py //各种个样dataset（paired unpaired combaind..)
transform.py //制作dataset的时候，用到的数据变化 比如norm crop等
utils.py //读取 加载 图片类型转化 padding等data用到的工具
model
vgg19.weight
models
auxiliary.py //作为特征提取或者vggloss 辅助网络
discriminators.py //组成鉴别器的不同组件 以及不同鉴别器
generators.py //组成生成器的不同组件 以及不同生成器
losses.py //比较复杂的loss 例如GAN perceptual loss
engan.py //实验的完整网络和对比网络 由G和D和loss共同构成
utils.py //初始化网络 初始化权重等 models部分用到的工具
outputs
实验输出
utils
metrics.py //一些评价指标
utils.py //系统时间啊 get device啊
visualize.py //图片可视化和保存
eval.py
train.py
test.py
README.md
|