Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
不会修仙的道士
labelme
提交
8ef581fe
L
labelme
项目概览
不会修仙的道士
/
labelme
与 Fork 源项目一致
从无法访问的项目Fork
通知
4
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
L
labelme
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
8ef581fe
编写于
3月 30, 2018
作者:
K
Kentaro Wada
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Update labelme_on_docker to handle input directory
上级
1febf239
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
32 addition
and
25 deletion
+32
-25
labelme/cli/on_docker.py
labelme/cli/on_docker.py
+32
-25
未找到文件。
labelme/cli/on_docker.py
浏览文件 @
8ef581fe
...
...
@@ -30,53 +30,60 @@ def get_ip():
raise
RuntimeError
(
'Unsupported platform.'
)
def
labelme_on_docker
(
i
mage
_file
,
out_file
):
def
labelme_on_docker
(
i
n
_file
,
out_file
):
ip
=
get_ip
()
cmd
=
'xhost + %s'
%
ip
subprocess
.
check_output
(
shlex
.
split
(
cmd
))
out_file
=
osp
.
abspath
(
out_file
)
if
osp
.
exists
(
out_file
):
raise
RuntimeError
(
'File exists: %s'
%
out_file
)
else
:
open
(
osp
.
abspath
(
out_file
),
'w'
)
if
out_file
:
out_file
=
osp
.
abspath
(
out_file
)
if
osp
.
exists
(
out_file
):
raise
RuntimeError
(
'File exists: %s'
%
out_file
)
else
:
open
(
osp
.
abspath
(
out_file
),
'w'
)
cmd
=
'docker run -it --rm'
\
' -e DISPLAY={0}:0'
\
' -e QT_X11_NO_MITSHM=1'
\
' -v /tmp/.X11-unix:/tmp/.X11-unix'
\
' -v {1}:{2}'
\
' -v {3}:{4}'
\
' -w /home/developer'
\
' wkentaro/labelme'
\
' labelme {2} -O {4}'
' -w /home/developer'
in_file_a
=
osp
.
abspath
(
in_file
)
in_file_b
=
osp
.
join
(
'/home/developer'
,
osp
.
basename
(
in_file
))
cmd
=
cmd
.
format
(
ip
,
osp
.
abspath
(
image_file
),
osp
.
join
(
'/home/developer'
,
osp
.
basename
(
image_file
)),
osp
.
abspath
(
out_file
),
osp
.
join
(
'/home/developer'
,
osp
.
basename
(
out_file
)),
in_file_a
,
in_file_b
,
)
if
out_file
:
out_file_a
=
osp
.
abspath
(
out_file
)
out_file_b
=
osp
.
join
(
'/home/developer'
,
osp
.
basename
(
out_file
))
cmd
+=
' -v {0}:{1}'
.
format
(
out_file_a
,
out_file_b
)
cmd
+=
' wkentaro/labelme labelme {0}'
.
format
(
in_file_b
)
if
out_file
:
cmd
+=
' -O {0}'
.
format
(
out_file_b
)
subprocess
.
call
(
shlex
.
split
(
cmd
))
try
:
json
.
load
(
open
(
out_file
))
return
out_file
except
Exception
as
e
:
if
open
(
out_file
).
read
()
==
''
:
os
.
remove
(
out_file
)
raise
RuntimeError
(
'Annotation is cancelled.'
)
if
out_file
:
try
:
json
.
load
(
open
(
out_file
))
return
out_file
except
Exception
as
e
:
if
open
(
out_file
).
read
()
==
''
:
os
.
remove
(
out_file
)
raise
RuntimeError
(
'Annotation is cancelled.'
)
def
main
():
parser
=
argparse
.
ArgumentParser
()
parser
.
add_argument
(
'i
mage_file
'
)
parser
.
add_argument
(
'-O'
,
'--output'
,
required
=
True
)
parser
.
add_argument
(
'i
n_file'
,
help
=
'Input file or directory.
'
)
parser
.
add_argument
(
'-O'
,
'--output'
)
args
=
parser
.
parse_args
()
try
:
out_file
=
labelme_on_docker
(
args
.
image_file
,
args
.
output
)
print
(
'Saved to: %s'
%
out_file
)
out_file
=
labelme_on_docker
(
args
.
in_file
,
args
.
output
)
if
out_file
:
print
(
'Saved to: %s'
%
out_file
)
except
RuntimeError
as
e
:
sys
.
stderr
.
write
(
e
.
__str__
()
+
'
\n
'
)
sys
.
exit
(
1
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录