Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
a123456jy
DeepMosaics
提交
27fffdb4
DeepMosaics
项目概览
a123456jy
/
DeepMosaics
与 Fork 源项目一致
Fork自
Hypo / DeepMosaics
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
DeepMosaics
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
27fffdb4
编写于
8月 16, 2019
作者:
H
hypox64
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
correct color
上级
6c6d0522
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
16 addition
and
4 deletion
+16
-4
deepmosaic.py
deepmosaic.py
+1
-1
options.py
options.py
+10
-2
util/image_processing.py
util/image_processing.py
+5
-1
未找到文件。
deepmosaic.py
浏览文件 @
27fffdb4
...
...
@@ -68,7 +68,7 @@ def main():
img_result
=
img_origin
.
copy
()
if
size
!=
0
:
img_mosaic
=
img_origin
[
y
-
size
:
y
+
size
,
x
-
size
:
x
+
size
]
img_fake
=
runmodel
.
run_pix2pix
(
img_mosaic
,
netG
,
opt
)
img_fake
=
runmodel
.
run_pix2pix
(
img_mosaic
,
netG
,
opt
)
img_result
=
impro
.
replace_mosaic
(
img_origin
,
img_fake
,
x
,
y
,
size
,
opt
.
no_feather
)
cv2
.
imwrite
(
os
.
path
.
join
(
opt
.
result_dir
,
os
.
path
.
basename
(
path
)),
img_result
)
...
...
options.py
浏览文件 @
27fffdb4
...
...
@@ -11,7 +11,7 @@ class Options():
#base
self
.
parser
.
add_argument
(
'--use_gpu'
,
action
=
'store_true'
,
help
=
'if input it, use gpu'
)
self
.
parser
.
add_argument
(
'--media_path'
,
type
=
str
,
default
=
'./hands_test.mp4'
,
help
=
'your videos or images path'
)
self
.
parser
.
add_argument
(
'--mode'
,
type
=
str
,
default
=
'a
dd'
,
help
=
'add or clean mosaic into your media
add | clean'
)
self
.
parser
.
add_argument
(
'--mode'
,
type
=
str
,
default
=
'a
uto'
,
help
=
'add or clean mosaic into your media auto |
add | clean'
)
self
.
parser
.
add_argument
(
'--model_path'
,
type
=
str
,
default
=
'./pretrained_models/add_hands_128.pth'
,
help
=
'pretrained model path'
)
self
.
parser
.
add_argument
(
'--result_dir'
,
type
=
str
,
default
=
'./result'
,
help
=
'output result will be saved here'
)
self
.
parser
.
add_argument
(
'--tempimage_type'
,
type
=
str
,
default
=
'png'
,
help
=
'type of temp image, png | jpg, png is better but occupy more storage space'
)
...
...
@@ -26,7 +26,7 @@ class Options():
#CleanMosaic
self
.
parser
.
add_argument
(
'--netG'
,
type
=
str
,
default
=
'auto'
,
help
=
'select model to use for netG(clean mosaic) -> auto | unet_128 | resnet_9blocks | HD'
)
self
.
parser
.
add_argument
(
'--mosaic_position_model_path'
,
type
=
str
,
default
=
'auto'
,
help
=
'name of model use to find mosaic position'
)
self
.
parser
.
add_argument
(
'--no_feather'
,
action
=
'store_true'
,
help
=
'if true, no edge feather
,
but run faster'
)
self
.
parser
.
add_argument
(
'--no_feather'
,
action
=
'store_true'
,
help
=
'if true, no edge feather
and color correction,
but run faster'
)
self
.
parser
.
add_argument
(
'--medfilt_num'
,
type
=
int
,
default
=
11
,
help
=
'medfilt window of mosaic movement in the video'
)
self
.
initialized
=
True
...
...
@@ -36,6 +36,14 @@ class Options():
self
.
initialize
()
self
.
opt
=
self
.
parser
.
parse_args
()
if
self
.
opt
.
mode
==
'auto'
:
if
'add'
in
self
.
opt
.
model_path
:
self
.
opt
.
mode
=
'add'
elif
'clean'
in
self
.
opt
.
model_path
:
self
.
opt
.
mode
=
'clean'
else
:
print
(
'Please input running mode!'
)
if
self
.
opt
.
netG
==
'auto'
and
self
.
opt
.
mode
==
'clean'
:
if
'unet_128'
in
self
.
opt
.
model_path
:
self
.
opt
.
netG
=
'unet_128'
...
...
util/image_processing.py
浏览文件 @
27fffdb4
...
...
@@ -124,11 +124,15 @@ def mask_area(mask):
def
replace_mosaic
(
img_origin
,
img_fake
,
x
,
y
,
size
,
no_father
):
img_fake
=
resize
(
img_fake
,
size
*
2
)
if
no_father
:
img_origin
[
y
-
size
:
y
+
size
,
x
-
size
:
x
+
size
]
=
img_fake
img_result
=
img_origin
else
:
#color correction
RGB_origin
=
img_origin
[
y
-
size
:
y
+
size
,
x
-
size
:
x
+
size
].
mean
(
0
).
mean
(
0
)
RGB_fake
=
img_fake
.
mean
(
0
).
mean
(
0
)
for
i
in
range
(
3
):
img_fake
[:,:,
i
]
=
np
.
clip
(
img_fake
[:,:,
i
]
+
RGB_origin
[
i
]
-
RGB_fake
[
i
],
0
,
255
)
#eclosion
eclosion_num
=
int
(
size
/
5
)
entad
=
int
(
eclosion_num
/
2
+
2
)
mask
=
np
.
zeros
(
img_origin
.
shape
,
dtype
=
'uint8'
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录