Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
models
提交
63d68a80
M
models
项目概览
PaddlePaddle
/
models
10 个月 前同步成功
通知
221
Star
6828
Fork
2962
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
602
列表
看板
标记
里程碑
合并请求
255
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
models
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
602
Issue
602
列表
看板
标记
里程碑
合并请求
255
合并请求
255
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
未验证
提交
63d68a80
编写于
7月 05, 2019
作者:
L
lvmengsi
提交者:
GitHub
7月 05, 2019
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Cherry pick gan (#2737)
Cherry pick gan
上级
ee491be3
变更
5
隐藏空白更改
内联
并排
Showing
5 changed file
with
37 addition
and
13 deletion
+37
-13
PaddleCV/PaddleGAN/README.md
PaddleCV/PaddleGAN/README.md
+4
-2
PaddleCV/PaddleGAN/infer.py
PaddleCV/PaddleGAN/infer.py
+8
-5
PaddleCV/PaddleGAN/train.py
PaddleCV/PaddleGAN/train.py
+1
-0
PaddleCV/PaddleGAN/trainer/StarGAN.py
PaddleCV/PaddleGAN/trainer/StarGAN.py
+0
-4
PaddleCV/PaddleGAN/util/utility.py
PaddleCV/PaddleGAN/util/utility.py
+24
-2
未找到文件。
PaddleCV/PaddleGAN/README.md
浏览文件 @
63d68a80
...
...
@@ -18,7 +18,9 @@
本图像生成模型库包含CGAN
\[
[
3
](
#参考文献
)
\]
, DCGAN
\[
[
4
](
#参考文献
)
\]
, Pix2Pix
\[
[
5
](
#参考文献
)
\]
, CycleGAN
\[
[
6
](
#参考文献
)
\]
, StarGAN
\[
[
7
](
#参考文献
)
\]
, AttGAN
\[
[
8
](
#参考文献
)
\]
, STGAN
\[
[
9
](
#参考文献
)
\]
。
注意:AttGAN和STGAN的网络结构中,判别器去掉了instance norm。
注意:
1.
AttGAN和STGAN的网络结构中,判别器去掉了instance norm。
2.
StarGAN,AttGAN和STGAN由于梯度惩罚所需的操作目前只支持GPU,需使用GPU训练。
图像生成模型库库的目录结构如下:
```
...
...
@@ -235,7 +237,7 @@ STGAN的网络结构[9]
**Q:**
为什么STGAN和ATTGAN中变男性得到的预测结果是变女性呢?
**A:**
这是由于预测时标签的设置,目标标签是基于原本的标签进行改变,比如原本图片是男生,预测代码对标签进行转变的时候会自动变成相对立的标签,即女
性,所以得到的结果是女生。如果想要原本是男生,转变之后还是男生,
可以参考模型库中预测代码的StarGAN的标签设置
。
性,所以得到的结果是女生。如果想要原本是男生,转变之后还是男生,
保持要转变的标签不变即可
。
## 参考论文
...
...
PaddleCV/PaddleGAN/infer.py
浏览文件 @
63d68a80
...
...
@@ -27,7 +27,7 @@ import imageio
import
glob
from
util.config
import
add_arguments
,
print_arguments
from
data_reader
import
celeba_reader_creator
from
util.utility
import
check_attribute_conflict
from
util.utility
import
check_attribute_conflict
,
check_gpu
import
copy
parser
=
argparse
.
ArgumentParser
(
description
=
__doc__
)
...
...
@@ -122,6 +122,8 @@ def infer(args):
if
not
os
.
path
.
exists
(
args
.
output
):
os
.
makedirs
(
args
.
output
)
attr_names
=
args
.
selected_attrs
.
split
(
','
)
if
args
.
model_net
==
'AttGAN'
or
args
.
model_net
==
'STGAN'
:
test_reader
=
celeba_reader_creator
(
image_dir
=
args
.
dataset_dir
,
...
...
@@ -133,7 +135,6 @@ def infer(args):
args
,
shuffle
=
False
,
return_name
=
True
)
for
data
in
zip
(
reader_test
()):
real_img
,
label_org
,
name
=
data
[
0
]
attr_names
=
args
.
selected_attrs
.
split
(
','
)
print
(
"read {}"
.
format
(
name
))
label_trg
=
copy
.
deepcopy
(
label_org
)
tensor_img
=
fluid
.
LoDTensor
()
...
...
@@ -189,10 +190,11 @@ def infer(args):
real_img_temp
=
np
.
squeeze
(
real_img
).
transpose
([
0
,
2
,
3
,
1
])
images
=
[
real_img_temp
]
for
i
in
range
(
args
.
c_dim
):
label_trg
=
np
.
zeros
(
[
len
(
label_org
),
args
.
c_dim
]).
astype
(
"float32"
)
label_trg_tmp
=
copy
.
deepcopy
(
label_org
)
for
j
in
range
(
len
(
label_org
)):
label_trg
[
j
][
i
]
=
1
label_trg_tmp
[
j
][
i
]
=
1.0
-
label_trg_tmp
[
j
][
i
]
label_trg
=
check_attribute_conflict
(
label_trg_tmp
,
attr_names
[
i
],
attr_names
)
tensor_label_trg
=
fluid
.
LoDTensor
()
tensor_label_trg
.
set
(
label_trg
,
place
)
out
=
exe
.
run
(
...
...
@@ -233,4 +235,5 @@ def infer(args):
if
__name__
==
"__main__"
:
args
=
parser
.
parse_args
()
print_arguments
(
args
)
check_gpu
(
args
.
use_gpu
)
infer
(
args
)
PaddleCV/PaddleGAN/train.py
浏览文件 @
63d68a80
...
...
@@ -76,6 +76,7 @@ def train(cfg):
if
__name__
==
"__main__"
:
cfg
=
config
.
parse_args
()
config
.
print_arguments
(
cfg
)
utility
.
check_gpu
(
cfg
.
use_gpu
)
#assert cfg.load_size >= cfg.crop_size, "Load Size CANNOT less than Crop Size!"
if
cfg
.
profile
:
if
cfg
.
use_gpu
:
...
...
PaddleCV/PaddleGAN/trainer/StarGAN.py
浏览文件 @
63d68a80
...
...
@@ -308,12 +308,8 @@ class StarGAN(object):
loss_name
=
dis_trainer
.
d_loss
.
name
,
build_strategy
=
build_strategy
)
#losses = [[], []]
t_time
=
0
test_program
=
gen_trainer
.
infer_program
utility
.
save_test_image
(
0
,
self
.
cfg
,
exe
,
place
,
test_program
,
gen_trainer
,
self
.
test_reader
)
for
epoch_id
in
range
(
self
.
cfg
.
epoch
):
batch_id
=
0
for
i
in
range
(
self
.
batch_num
):
...
...
PaddleCV/PaddleGAN/util/utility.py
浏览文件 @
63d68a80
...
...
@@ -104,6 +104,7 @@ def save_test_image(epoch,
elif
cfg
.
model_net
==
"StarGAN"
:
for
data
in
zip
(
A_test_reader
()):
real_img
,
label_org
,
name
=
data
[
0
]
attr_names
=
cfg
.
selected_attrs
.
split
(
','
)
tensor_img
=
fluid
.
LoDTensor
()
tensor_label_org
=
fluid
.
LoDTensor
()
tensor_img
.
set
(
real_img
,
place
)
...
...
@@ -111,8 +112,10 @@ def save_test_image(epoch,
real_img_temp
=
np
.
squeeze
(
real_img
).
transpose
([
1
,
2
,
0
])
images
=
[
real_img_temp
]
for
i
in
range
(
cfg
.
c_dim
):
label_trg
=
np
.
zeros
([
1
,
cfg
.
c_dim
]).
astype
(
"float32"
)
label_trg
[
0
][
i
]
=
1
label_trg_tmp
=
copy
.
deepcopy
(
label_org
)
label_trg_tmp
[
0
][
i
]
=
1.0
-
label_trg_tmp
[
j
][
i
]
label_trg
=
check_attribute_conflict
(
label_trg_tmp
,
attr_names
[
i
],
attr_names
)
tensor_label_trg
=
fluid
.
LoDTensor
()
tensor_label_trg
.
set
(
label_trg
,
place
)
fake_temp
,
rec_temp
=
exe
.
run
(
...
...
@@ -261,3 +264,22 @@ def check_attribute_conflict(label_batch, attr, attrs):
if
a
!=
attr
:
_set
(
label
,
0
,
a
)
return
label_batch
def
check_gpu
(
use_gpu
):
"""
Log error and exit when set use_gpu=true in paddlepaddle
cpu version.
"""
err
=
"Config use_gpu cannot be set as true while you are "
\
"using paddlepaddle cpu version !
\n
Please try:
\n
"
\
"
\t
1. Install paddlepaddle-gpu to run model on GPU
\n
"
\
"
\t
2. Set use_gpu as false in config file to run "
\
"model on CPU"
try
:
if
use_gpu
and
not
fluid
.
is_compiled_with_cuda
():
logger
.
error
(
err
)
sys
.
exit
(
1
)
except
Exception
as
e
:
pass
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录