python标注制作_使用labelme软件标注并制作自己的数据集

### Labelme 库中 `shapes_to_label` 和 `lblsave` 函数的使用 #### shapes_to_label 方法解析 `labelme.utils.shapes_to_label` 是用于将由Labelme生成的JSON文件中的形状（多边形、矩形等）转换成标签掩码图像的方法。此方法接受多个参数，包括图像尺寸、形状列表以及类别名称到ID映射字典。 ```python import numpy as np from labelme import utils def convert_shapes_to_labels(img_shape, shapes, label_name_to_value): lbl, _ = utils.shapes_to_label( img_shape=img_shape, shapes=shapes, label_name_to_value=label_name_to_value ) return lbl ``` 上述代码展示了如何定义一个辅助函数来调用 `utils.shapes_to_label()` 并返回标签数组 `lbl`[^1]。 #### lblsave 函数介绍 `labelme.utils.lblsave` 则是用来保存标签掩码图像至磁盘上的PNG文件。该功能可以方便地把处理后的二值化或彩色编码的结果存储下来以便后续查看或者作为模型输入的一部分。 ```python from PIL import Image import os.path as osp from labelme import utils def save_label_as_png(filename, lbl): if osp.splitext(filename)[1] != '.png': filename += '.png' # Ensure the directory exists. dirname = osp.dirname(filename) if not osp.exists(dirname): os.makedirs(dirname) utils.lblsave(filename, lbl.astype(np.int32)) ``` 这段脚本说明了怎样创建一个新的实用程序函数 `save_label_as_png()`, 它接收目标路径名和标签矩阵两个参数，并利用 `utils.lblsave()` 将其写入指定位置。 #### 实际应用案例展示 下面给出一段完整的Python代码片段，它读取了一个LabelMe JSON文件并将其转化为对应的标签图再保存： ```python import json import cv2 import numpy as np from pathlib import Path from labelme import utils json_file = './example.json' # 替换成实际存在的Labelme标注文件路径 output_dir = './converted/' # 输出目录 with open(json_file) as f: data = json.load(f) img_data = cv2.imread(str(Path(json_file).parent / data['imagePath'])) height, width = img_data.shape[:2] # 构建从类名到索引编号之间的对应关系表 label_names = ['__background__'] + [shape['label'] for shape in set([tuple(shape.items()) for shape in data['shapes']])] label_values = list(range(len(label_names))) label_map = dict(zip(label_names, label_values)) # 调用API完成转化工作 label_img = utils.shapes_to_label((height, width), data['shapes'], label_map) # 存储结果 utils.lblsave(output_dir + 'result.png', label_img) print('Conversion completed.') ``` 以上示例实现了从加载原始标注数据到最后一步输出最终标签图像的过程。

使用 labelme 软件和在 Python 中打开 labelme 进行标注的主要区别在于使用的方式不同。使用 labelme 软件，您可以直接在界面上标注图像数据，然后将结果保存为 JSON 文件。而在 Python 中打开 labelme，您需要编写代码来加载图像和标注数据，然后在代码中完成标注并将结果保存为 JSON 文件。 如果您只需要进行简单的标注操作，那么使用 labelme 软件可能更加方便和直观。但是，如果您需要将标注与其他 Python 代码集成，或者需要进行批量标注等高级操作，那么在 Python 中打开 labelme 可能更加灵活和方便。 因此，更推荐的方法取决于您的具体需求。如果您只需要进行简单的标注操作，可以使用 labelme 软件。如果您需要进行更高级的操作，或者需要与其他 Python 代码集成，可以使用在 Python 中打开 labelme。