关于json_to_dataset时出现 module ‘labelme.utils‘ has no attribute ‘draw_label‘的解决

关于json_to_dataset时出现 module ‘labelme.utils’ has no attribute ‘draw_label问题解决’

相信肯定很多朋友遇到过这种问题：

AttributeError: module ' labelme. utils' has no attribute ' draw label'

终端显示模块utils缺少属性draw_label.

其实呢，这个问题我弄了两天都没有解决，最后我决定放弃这个函数。
之后怎么办呢，相信出现这种问题时大家都是从别人博客里copy过json_to_dataset这个文件的代码的。既然别人的代码修改时避不开draw_label，我选择回到最开始。

直接修改label的json_to_dataset文件初始代码

你可能会说，以前的代码已经没了，不要紧，在终端卸掉labelme再重新安装就好了。
以下是我修改的代码：

import argparse
import base64
import json
import os
import os.path as osp

import imgviz
import PIL.Image

from labelme.logger import logger
from labelme import utils
import cv2
from math import *
import numpy as np
import random



def main():
    list_path = os.listdir('F:/zkx')      # 这是我的路径，替换成自己的就好
    for i in range(0, len(list_path)):
            logger.warning('This script is aimed to demonstrate how to convert the'
                           'JSON file to a single image dataset, and not to handle'
                           'multiple JSON files to generate a real-use dataset.')

            parser = argparse.ArgumentParser()
            parser.add_argument('--json_file')
            parser.add_argument('-o', '--out', default=None)
            args = parser.parse_args()

            json_file = 'F:/zkx/' + list_path[i]         # 这是我的路径，替换成自己的就好
            print(list_path[i])
            if args.out is None:
                out_dir = osp.basename(json_file).replace('.', '_')  # 返回文件名
                out_dir = osp.join(osp.dirname(json_file), out_dir)  # 把目录和文件名合成一个路径
            else:
                out_dir = args.out
            if not osp.exists(out_dir):
                os.mkdir(out_dir)  # 用于以数字权限模式创建目录

            data = json.load(open(json_file))
            imageData = data.get('imageData')

            if not imageData:
                imagePath = os.path.join(os.path.dirname(json_file), data['imagePath']) # os.path.dirname返回文件路径
                with open(imagePath, 'rb') as f:
                    imageData = f.read()
                    imageData = base64.b64encode(imageData).decode('utf-8')
            img = utils.img_b64_to_arr(imageData)

            label_name_to_value = {'_background_': 0}
            for shape in sorted(data['shapes'], key=lambda x: x['label']):
                label_name = shape['label']
                if label_name in label_name_to_value:
                    label_value = label_name_to_value[label_name]
                else:
                    label_value = len(label_name_to_value)
                    label_name_to_value[label_name] = label_value
            lbl, _ = utils.shapes_to_label(
                img.shape, data['shapes'], label_name_to_value
            )

            label_names = [None] * (max(label_name_to_value.values()) + 1)
            for name, value in label_name_to_value.items():
                label_names[value] = name

            lbl_viz = imgviz.label2rgb(
                label=lbl, img=imgviz.asgray(img), label_names=label_names, loc='rb'
            )

            PIL.Image.fromarray(img).save(osp.join(out_dir, 'img.png'))
            utils.lblsave(osp.join(out_dir, 'label.png'), lbl)
            PIL.Image.fromarray(lbl_viz).save(osp.join(out_dir, 'label_viz.png'))

            with open(osp.join(out_dir, 'label_names.txt'), 'w') as f:
                for lbl_name in label_names:
                    f.write(lbl_name + '\n')

            logger.info('Saved to: {}'.format(out_dir))
            x = out_dir+'\\label.png'


if __name__ == '__main__':
    main()
    

执行的脚本，结果如下：
每个文件夹内包含的文件如下：

[1]:https://blog.youkuaiyun.com/szumaine/article/details/104407182