使用labelme批量注释的时候出现：module 'labelme.utils' has no attribute 'label_colormap'或者提示没有 'draw_label'

最新推荐文章于 2024-01-08 15:16:34 发布

旋涡小林

最新推荐文章于 2024-01-08 15:16:34 发布

阅读量4k

点赞数 5

分类专栏：那些年踩过的坑文章标签：经验分享

本文链接：https://blog.csdn.net/gyyu32g/article/details/104725435

版权

在使用labelme进行批量标注时遇到错误，提示缺少'draw_label'或'label_colormap'属性。问题源于缺少draw.py文件。解决方案包括检查anaconda虚拟环境的labelme/utils目录，确认并添加draw.py文件，同时在__init__.py中导入相关模块。

摘要由CSDN通过智能技术生成

1 问题

在使用labelme进行批量标注的时候出现如下错误：

module 'labelme.utils' has no attribute 'label_colormap'

或者：

module 'labelme.utils' has no attribute  'draw_label'

2 产生的原因以及解决方法

出现以上的两个问题中的一个或者两个，其都是因为安装的labelme缺少了一个draw.py文件，可以到你的anaconda路径下的 \envs\你的虚拟环境的名字\Lib\site-packages\labelme\utils 目录下查看是否缺失了draw.py文件：在这里插入图片描述
如果缺失了这个文件，说明就是由于缺失这个文件导致缺失了这个模块。可以这样做：

① 可以在这个目录下创建draw.py文件，然后复制下面的代码进去。

import io

import numpy as np
import PIL.Image
import PIL.ImageDraw


def label_colormap(N=256):

    def bitget(byteval, idx):
        return ((byteval & (1 << idx)) != 0)

    cmap = np.zeros((N, 3))
    for i in range(0, N):
        id = i
        r, g, b = 0, 0, 0
        for j in range(0, 8):
            r = np.bitwise_or(r, (bitget(id, 0) << 7 - j))
            g = np.bitwise_or(g