提交 c6266dcf 编写于 作者: K Kentaro Wada
上级 61f5fee4
...@@ -56,8 +56,9 @@ def main(): ...@@ -56,8 +56,9 @@ def main():
label_names = [None] * (max(label_name_to_value.values()) + 1) label_names = [None] * (max(label_name_to_value.values()) + 1)
for name, value in label_name_to_value.items(): for name, value in label_name_to_value.items():
label_names[value] = name label_names[value] = name
lbl_viz = imgviz.label2rgb( lbl_viz = imgviz.label2rgb(
label=lbl, img=imgviz.rgb2gray(img), label_names=label_names, loc='rb' label=lbl, img=imgviz.asgray(img), label_names=label_names, loc='rb'
) )
PIL.Image.fromarray(img).save(osp.join(out_dir, 'img.png')) PIL.Image.fromarray(img).save(osp.join(out_dir, 'img.png'))
......
...@@ -27,7 +27,7 @@ def get_install_requires(): ...@@ -27,7 +27,7 @@ def get_install_requires():
assert PY3 or PY2 assert PY3 or PY2
install_requires = [ install_requires = [
'imgviz>=0.10.6', 'imgviz>=0.11.0',
'matplotlib', 'matplotlib',
'numpy', 'numpy',
'Pillow>=2.8.0', 'Pillow>=2.8.0',
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册