Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
xiaoqin0241
PaddleGAN
提交
52cc814f
P
PaddleGAN
项目概览
xiaoqin0241
/
PaddleGAN
与 Fork 源项目一致
Fork自
PaddlePaddle / PaddleGAN
通知
1
Star
1
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleGAN
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
52cc814f
编写于
10月 13, 2020
作者:
L
LielinJiang
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
clean code
上级
e8f33934
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
8 addition
and
7 deletion
+8
-7
applications/tools/video-enhance.py
applications/tools/video-enhance.py
+2
-6
ppgan/apps/__init__.py
ppgan/apps/__init__.py
+5
-0
ppgan/apps/deepremaster_predictor.py
ppgan/apps/deepremaster_predictor.py
+1
-1
未找到文件。
applications/tools/video-enhance.py
浏览文件 @
52cc814f
...
@@ -15,11 +15,7 @@
...
@@ -15,11 +15,7 @@
import
argparse
import
argparse
import
paddle
import
paddle
from
ppgan.apps.dain_predictor
import
DAINPredictor
from
ppgan.apps
import
DAINPredictor
,
DeepRemasterPredictor
,
DeOldifyPredictor
,
RealSRPredictor
,
EDVRPredictor
from
ppgan.apps.deepremaster_predictor
import
DeepReasterPredictor
from
ppgan.apps.deoldify_predictor
import
DeOldifyPredictor
from
ppgan.apps.realsr_predictor
import
RealSRPredictor
from
ppgan.apps.edvr_predictor
import
EDVRPredictor
parser
=
argparse
.
ArgumentParser
(
description
=
'Fix video'
)
parser
=
argparse
.
ArgumentParser
(
description
=
'Fix video'
)
parser
.
add_argument
(
'--input'
,
type
=
str
,
default
=
None
,
help
=
'Input video'
)
parser
.
add_argument
(
'--input'
,
type
=
str
,
default
=
None
,
help
=
'Input video'
)
...
@@ -97,7 +93,7 @@ if __name__ == "__main__":
...
@@ -97,7 +93,7 @@ if __name__ == "__main__":
frames_path
,
temp_video_path
=
predictor
.
run
(
temp_video_path
)
frames_path
,
temp_video_path
=
predictor
.
run
(
temp_video_path
)
paddle
.
disable_static
()
paddle
.
disable_static
()
elif
order
==
'DeepRemaster'
:
elif
order
==
'DeepRemaster'
:
predictor
=
DeepReasterPredictor
(
predictor
=
DeepRe
m
asterPredictor
(
args
.
output
,
args
.
output
,
weight_path
=
args
.
DeepRemaster_weight
,
weight_path
=
args
.
DeepRemaster_weight
,
colorization
=
args
.
colorization
,
colorization
=
args
.
colorization
,
...
...
ppgan/apps/__init__.py
0 → 100644
浏览文件 @
52cc814f
from
.dain_predictor
import
DAINPredictor
from
.deepremaster_predictor
import
DeepRemasterPredictor
from
.deoldify_predictor
import
DeOldifyPredictor
from
.realsr_predictor
import
RealSRPredictor
from
.edvr_predictor
import
EDVRPredictor
ppgan/apps/deepremaster_predictor.py
浏览文件 @
52cc814f
...
@@ -64,7 +64,7 @@ def addMergin(img, target_w, target_h, background_color=(0, 0, 0)):
...
@@ -64,7 +64,7 @@ def addMergin(img, target_w, target_h, background_color=(0, 0, 0)):
return
result
return
result
class
DeepReasterPredictor
(
BasePredictor
):
class
DeepRe
m
asterPredictor
(
BasePredictor
):
def
__init__
(
self
,
def
__init__
(
self
,
output
=
'output'
,
output
=
'output'
,
weight_path
=
None
,
weight_path
=
None
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录