Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
无所谓斗争
DeepMosaics
提交
347fb5ca
DeepMosaics
项目概览
无所谓斗争
/
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,发现更多精彩内容 >>
提交
347fb5ca
编写于
7月 29, 2019
作者:
H
hypox64
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Modified file system
上级
33ceac66
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
28 addition
and
17 deletion
+28
-17
deepmosaic.py
deepmosaic.py
+5
-3
util/ffmpeg.py
util/ffmpeg.py
+2
-1
util/util.py
util/util.py
+21
-13
未找到文件。
deepmosaic.py
浏览文件 @
347fb5ca
...
...
@@ -13,7 +13,7 @@ from options import Options
opt
=
Options
().
getparse
()
util
.
init
(
opt
)
util
.
file_
init
(
opt
)
if
opt
.
mode
==
'add'
:
...
...
@@ -29,7 +29,9 @@ if opt.mode == 'add':
fps
=
ffmpeg
.
get_video_infos
(
path
)[
0
]
ffmpeg
.
video2voice
(
path
,
'./tmp/voice_tmp.mp3'
)
ffmpeg
.
video2image
(
path
,
'./tmp/video2image/output_%05d.'
+
opt
.
tempimage_type
)
for
imagepath
in
os
.
listdir
(
'./tmp/video2image'
):
imagepaths
=
os
.
listdir
(
'./tmp/video2image'
)
imagepaths
.
sort
()
for
imagepath
in
imagepaths
:
imagepath
=
os
.
path
.
join
(
'./tmp/video2image'
,
imagepath
)
print
(
'Add Mosaic:'
,
imagepath
)
img
=
cv2
.
imread
(
imagepath
)
...
...
@@ -90,4 +92,4 @@ elif opt.mode == 'clean':
'./tmp/voice_tmp.mp3'
,
os
.
path
.
join
(
opt
.
result_dir
,
os
.
path
.
splitext
(
os
.
path
.
basename
(
path
))[
0
]
+
'_CleanMosaic.mp4'
))
util
.
clean_tempfiles
()
\ No newline at end of file
util
.
clean_tempfiles
(
tmp_init
=
False
)
\ No newline at end of file
util/ffmpeg.py
浏览文件 @
347fb5ca
...
...
@@ -7,7 +7,8 @@ def video2voice(videopath,voicepath):
os
.
system
(
'ffmpeg -i '
+
videopath
+
' -f mp3 '
+
voicepath
)
def
image2video
(
fps
,
imagepath
,
voicepath
,
videopath
):
os
.
system
(
'ffmpeg -y -r '
+
str
(
fps
)
+
' -i '
+
imagepath
+
' -vcodec libx264 '
+
'./tmp/video_tmp.mp4'
)
#os.system('ffmpeg -y -r '+str(fps)+' -i '+imagepath+' -vcodec libx264 '+'./tmp/video_tmp.mp4')
os
.
system
(
'ffmpeg -f image2 -i '
+
imagepath
+
' -vcodec libx264 -r '
+
str
(
fps
)
+
' ./tmp/video_tmp.mp4'
)
os
.
system
(
'ffmpeg -i ./tmp/video_tmp.mp4 -i '
+
voicepath
+
' -vcodec copy -acodec copy '
+
videopath
)
def
get_video_infos
(
videopath
):
...
...
util/util.py
浏览文件 @
347fb5ca
import
os
import
shutil
def
Traversal
(
filedir
):
file_list
=
[]
for
root
,
dirs
,
files
in
os
.
walk
(
filedir
):
...
...
@@ -24,21 +24,29 @@ def is_video(path):
return
True
else
:
return
False
def
init
(
opt
):
if
not
os
.
path
.
isdir
(
opt
.
result_dir
):
os
.
makedirs
(
opt
.
result_dir
)
print
(
'makedir:'
,
opt
.
result_dir
)
def
writelog
(
path
,
log
):
f
=
open
(
path
,
'a+'
)
f
.
write
(
log
+
'
\n
'
)
def
clean_tempfiles
():
# def del_all(dir_path):
# files = Traversal(dir_path)
# for file in files:
# os.remove(file)
# os.removedirs(dir_path)
def
clean_tempfiles
(
tmp_init
=
True
):
if
os
.
path
.
isdir
(
'./tmp'
):
os
.
system
(
'rm -rf ./tmp'
)
os
.
makedirs
(
'./tmp'
)
os
.
makedirs
(
'./tmp/video2image'
)
os
.
makedirs
(
'./tmp/addmosaic_image'
)
os
.
makedirs
(
'./tmp/mosaic_crop'
)
os
.
makedirs
(
'./tmp/replace_mosaic'
)
\ No newline at end of file
shutil
.
rmtree
(
'./tmp'
)
if
tmp_init
:
os
.
makedirs
(
'./tmp'
)
os
.
makedirs
(
'./tmp/video2image'
)
os
.
makedirs
(
'./tmp/addmosaic_image'
)
os
.
makedirs
(
'./tmp/mosaic_crop'
)
os
.
makedirs
(
'./tmp/replace_mosaic'
)
def
file_init
(
opt
):
if
not
os
.
path
.
isdir
(
opt
.
result_dir
):
os
.
makedirs
(
opt
.
result_dir
)
print
(
'makedir:'
,
opt
.
result_dir
)
clean_tempfiles
()
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录