提交 68fb3ebc 编写于 作者: J Jongha Lim 提交者: Kentaro Wada

Replace labelme.cli.draw_label_png with imgviz

上级 a8db0cfb
import argparse
import imgviz
import matplotlib.pyplot as plt
import numpy as np
import PIL.Image
from labelme.logger import logger
from labelme import utils
def main():
......@@ -19,7 +19,7 @@ def main():
logger.info('label shape: {}'.format(lbl.shape))
logger.info('unique label values: {}'.format(np.unique(lbl)))
lbl_viz = utils.draw_label(lbl)
lbl_viz = imgviz.label2rgb(lbl)
plt.imshow(lbl_viz)
plt.show()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册