从零到一:手把手教你用Labelme打造专属Mask数据集

news2026/4/11 19:00:51
1. 为什么你需要掌握Labelme标注工具如果你正在做计算机视觉相关的项目尤其是图像分割任务那么高质量的数据集就是你的命根子。市面上现成的数据集往往无法完全匹配你的业务需求这时候自己动手标注数据就成了必经之路。我刚开始接触图像分割时也走过不少弯路直到发现了Labelme这个神器。Labelme是麻省理工学院开发的一款开源图像标注工具特别适合做语义分割和实例分割的数据标注。它的最大优势就是简单易用哪怕你完全没有编程基础也能快速上手。我见过不少团队花大价钱购买商业标注工具其实对于大多数中小项目来说Labelme完全够用。相比其他标注工具Labelme有这几个杀手锏完全免费开源、支持多边形标注、输出格式通用JSON、跨平台支持。最重要的是它生成的标注文件可以很方便地转换为Mask图像这正是训练分割模型最需要的数据格式。2. 环境准备从零搭建标注工作站2.1 安装Labelme的三种方式很多人卡在第一步的安装环节其实Labelme的安装比你想象中简单得多。根据你的使用习惯我推荐三种安装方式直接下载可执行文件最适合小白 去Labelme的GitHub发布页下载对应系统的.exe或.dmg文件解压就能用。这种方式最省心但可能不是最新版本。通过Anaconda安装推荐给Python用户conda create --namelabelme python3.8 conda activate labelme pip install labelme这种方式能保证你获得最新版本而且可以和其他Python工具链完美配合。从源码安装适合开发者git clone https://github.com/wkentaro/labelme.git cd labelme pip install -e .我个人的建议是如果你只是偶尔用用选第一种如果要长期使用第二种是最佳选择除非你要修改源码否则没必要选第三种。2.2 准备你的标注素材在启动Labelme之前先整理好你的图片素材。这里有几个实用建议图片命名要有规律比如001.jpg、002.jpg这样顺序编号图片尺寸不要太大建议长宽都在1000像素以内最好把所有图片放在同一个文件夹内如果是团队协作建议先制定好标注规范比如哪些类别要标注我见过不少新手犯的一个错误是图片尺寸不统一有的横屏有的竖屏这会给后续处理带来不必要的麻烦。建议先用批量处理工具把图片调整为统一尺寸。3. 手把手教你标注第一张图片3.1 启动Labelme并加载图片安装完成后打开Labelme如果是命令行安装的直接输入labelme命令。你会看到一个简洁的界面点击Open按钮选择你的图片文件夹使用左右箭头键可以在图片间切换右下角可以调整图片显示大小第一次使用时我建议你先熟悉这几个快捷键Ctrl鼠标滚轮缩放图片空格键拖动图片CtrlS快速保存3.2 开始多边形标注假设我们要标注图片中的一只猫操作步骤如下点击左侧工具栏的Create Polygon按钮或者按快捷键P沿着猫的轮廓依次点击形成多边形节点遇到曲线部分可以多设几个点完成一圈后按Enter键闭合多边形在弹出的对话框中输入标签名称比如cat这里有几个实用技巧按住Ctrl键可以微调节点位置右键点击节点可以删除标注复杂物体时可以先用大块多边形勾勒大致形状再逐步细化我刚开始标注时总想一次到位结果效率很低。后来发现分阶段标注更高效先粗略标注整体再逐步添加细节。3.3 保存标注结果标注完成后点击Save按钮或CtrlS保存为JSON文件。Labelme会自动生成一个与图片同名的.json文件里面包含了所有标注的多边形坐标信息。重要提示建议专门创建一个annotations文件夹存放这些JSON文件与原始图片分开管理。这样后续处理会更方便。4. 将JSON转换为Mask图像4.1 理解Labelme的输出格式Labelme生成的JSON文件其实是一个结构化的标注数据包含了图片路径和尺寸信息每个标注多边形的顶点坐标对应的类别标签要将其转换为模型训练可用的Mask图像我们需要提取这些多边形信息并用纯色填充。4.2 使用Python脚本转换下面是我优化过的转换脚本比网上常见的版本更健壮import json import numpy as np import cv2 import os from pathlib import Path def json_to_mask(json_path, output_dir): # 读取JSON文件 with open(json_path, r) as f: data json.load(f) # 读取原始图片获取尺寸 img_path str(Path(json_path).parent / data[imagePath]) image cv2.imread(img_path) if image is None: raise ValueError(f无法读取图片: {img_path}) # 创建空白Mask mask np.zeros_like(image, dtypenp.uint8) # 处理每个标注形状 for shape in data[shapes]: points np.array(shape[points], dtypenp.int32) # 不同类别使用不同颜色 if shape[label] cat: color (255, 0, 0) # 红色代表猫 elif shape[label] dog: color (0, 255, 0) # 绿色代表狗 else: color (0, 0, 255) # 蓝色代表其他 cv2.fillPoly(mask, [points], color) # 保存Mask图像 os.makedirs(output_dir, exist_okTrue) mask_name f{Path(json_path).stem}_mask.png cv2.imwrite(str(Path(output_dir) / mask_name), mask) print(f已保存: {mask_name}) # 示例用法 json_to_mask(501.json, masks)这个脚本做了几点改进自动处理图片路径问题增加了错误处理支持不同类别用不同颜色标注自动创建输出目录4.3 批量转换技巧如果你有很多JSON文件需要转换可以用这个批量处理脚本import glob json_files glob.glob(annotations/*.json) for json_file in json_files: try: json_to_mask(json_file, masks) except Exception as e: print(f处理{json_file}时出错: {str(e)}) continue把这段代码保存为batch_convert.py然后运行即可一键转换整个文件夹的标注文件。5. 高级技巧与常见问题排查5.1 提高标注效率的秘诀经过多个项目的实践我总结出这几个提升标注效率的方法使用快捷键P快速开始多边形标注D复制上一个标注的形状CtrlZ撤销上一步操作分层标注法 先标注大致的形状再用Edit Polygon工具细化边缘。我通常分三轮标注第一轮快速标注所有目标的大致轮廓第二轮调整明显不准确的部分第三轮精细调整边缘细节团队协作技巧 如果是多人标注一定要先制定统一的标注规范包括哪些物体需要标注标注的精细程度标签命名规则特殊情况处理方式5.2 常见问题解决方案问题1标注时鼠标卡顿解决方案降低图片显示质量在View菜单中勾选Reduce image quality for faster drawing问题2转换后的Mask有空洞原因多边形自相交解决方案在标注时避免线段交叉或者使用cv2.fillPoly的lineTypecv2.LINE_AA参数问题3JSON文件损坏预防措施定期备份标注文件修复方法尝试用文本编辑器打开JSON文件检查是否有明显错误问题4标注结果不精确解决方案放大图片局部Ctrl鼠标滚轮进行精细标注或者使用Edit Polygon工具调整已有标注5.3 与其他工具的配合使用Labelme生成的Mask可以很方便地用于各种深度学习框架。这里给出两个典型用例PyTorch数据加载from torch.utils.data import Dataset class SegmentationDataset(Dataset): def __init__(self, image_dir, mask_dir): self.image_paths sorted(Path(image_dir).glob(*.jpg)) self.mask_paths sorted(Path(mask_dir).glob(*.png)) def __getitem__(self, idx): image cv2.imread(str(self.image_paths[idx])) mask cv2.imread(str(self.mask_paths[idx]), 0) # 灰度模式 return image, mask数据增强处理import albumentations as A transform A.Compose([ A.RandomRotate90(), A.Flip(), A.RandomBrightnessContrast(), ])在实际项目中我通常会先用Labelme标注100-200张样本训练一个基础模型然后用这个模型辅助标注剩下的图片可以大幅提升效率。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2507097.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

SpringBoot-17-MyBatis动态SQL标签之常用标签

文章目录 1 代码1.1 实体User.java1.2 接口UserMapper.java1.3 映射UserMapper.xml1.3.1 标签if1.3.2 标签if和where1.3.3 标签choose和when和otherwise1.4 UserController.java2 常用动态SQL标签2.1 标签set2.1.1 UserMapper.java2.1.2 UserMapper.xml2.1.3 UserController.ja…

wordpress后台更新后 前端没变化的解决方法

使用siteground主机的wordpress网站,会出现更新了网站内容和修改了php模板文件、js文件、css文件、图片文件后,网站没有变化的情况。 不熟悉siteground主机的新手,遇到这个问题,就很抓狂,明明是哪都没操作错误&#x…

网络编程(Modbus进阶)

思维导图 Modbus RTU(先学一点理论) 概念 Modbus RTU 是工业自动化领域 最广泛应用的串行通信协议,由 Modicon 公司(现施耐德电气)于 1979 年推出。它以 高效率、强健性、易实现的特点成为工业控制系统的通信标准。 包…

UE5 学习系列(二)用户操作界面及介绍

这篇博客是 UE5 学习系列博客的第二篇,在第一篇的基础上展开这篇内容。博客参考的 B 站视频资料和第一篇的链接如下: 【Note】:如果你已经完成安装等操作,可以只执行第一篇博客中 2. 新建一个空白游戏项目 章节操作,重…

IDEA运行Tomcat出现乱码问题解决汇总

最近正值期末周,有很多同学在写期末Java web作业时,运行tomcat出现乱码问题,经过多次解决与研究,我做了如下整理: 原因: IDEA本身编码与tomcat的编码与Windows编码不同导致,Windows 系统控制台…

利用最小二乘法找圆心和半径

#include <iostream> #include <vector> #include <cmath> #include <Eigen/Dense> // 需安装Eigen库用于矩阵运算 // 定义点结构 struct Point { double x, y; Point(double x_, double y_) : x(x_), y(y_) {} }; // 最小二乘法求圆心和半径 …

使用docker在3台服务器上搭建基于redis 6.x的一主两从三台均是哨兵模式

一、环境及版本说明 如果服务器已经安装了docker,则忽略此步骤,如果没有安装,则可以按照一下方式安装: 1. 在线安装(有互联网环境): 请看我这篇文章 传送阵>> 点我查看 2. 离线安装(内网环境):请看我这篇文章 传送阵>> 点我查看 说明&#xff1a;假设每台服务器已…

XML Group端口详解

在XML数据映射过程中&#xff0c;经常需要对数据进行分组聚合操作。例如&#xff0c;当处理包含多个物料明细的XML文件时&#xff0c;可能需要将相同物料号的明细归为一组&#xff0c;或对相同物料号的数量进行求和计算。传统实现方式通常需要编写脚本代码&#xff0c;增加了开…

LBE-LEX系列工业语音播放器|预警播报器|喇叭蜂鸣器的上位机配置操作说明

LBE-LEX系列工业语音播放器|预警播报器|喇叭蜂鸣器专为工业环境精心打造&#xff0c;完美适配AGV和无人叉车。同时&#xff0c;集成以太网与语音合成技术&#xff0c;为各类高级系统&#xff08;如MES、调度系统、库位管理、立库等&#xff09;提供高效便捷的语音交互体验。 L…

(LeetCode 每日一题) 3442. 奇偶频次间的最大差值 I (哈希、字符串)

题目&#xff1a;3442. 奇偶频次间的最大差值 I 思路 &#xff1a;哈希&#xff0c;时间复杂度0(n)。 用哈希表来记录每个字符串中字符的分布情况&#xff0c;哈希表这里用数组即可实现。 C版本&#xff1a; class Solution { public:int maxDifference(string s) {int a[26]…

【大模型RAG】拍照搜题技术架构速览:三层管道、两级检索、兜底大模型

摘要 拍照搜题系统采用“三层管道&#xff08;多模态 OCR → 语义检索 → 答案渲染&#xff09;、两级检索&#xff08;倒排 BM25 向量 HNSW&#xff09;并以大语言模型兜底”的整体框架&#xff1a; 多模态 OCR 层 将题目图片经过超分、去噪、倾斜校正后&#xff0c;分别用…

【Axure高保真原型】引导弹窗

今天和大家中分享引导弹窗的原型模板&#xff0c;载入页面后&#xff0c;会显示引导弹窗&#xff0c;适用于引导用户使用页面&#xff0c;点击完成后&#xff0c;会显示下一个引导弹窗&#xff0c;直至最后一个引导弹窗完成后进入首页。具体效果可以点击下方视频观看或打开下方…

接口测试中缓存处理策略

在接口测试中&#xff0c;缓存处理策略是一个关键环节&#xff0c;直接影响测试结果的准确性和可靠性。合理的缓存处理策略能够确保测试环境的一致性&#xff0c;避免因缓存数据导致的测试偏差。以下是接口测试中常见的缓存处理策略及其详细说明&#xff1a; 一、缓存处理的核…

龙虎榜——20250610

上证指数放量收阴线&#xff0c;个股多数下跌&#xff0c;盘中受消息影响大幅波动。 深证指数放量收阴线形成顶分型&#xff0c;指数短线有调整的需求&#xff0c;大概需要一两天。 2025年6月10日龙虎榜行业方向分析 1. 金融科技 代表标的&#xff1a;御银股份、雄帝科技 驱动…

观成科技:隐蔽隧道工具Ligolo-ng加密流量分析

1.工具介绍 Ligolo-ng是一款由go编写的高效隧道工具&#xff0c;该工具基于TUN接口实现其功能&#xff0c;利用反向TCP/TLS连接建立一条隐蔽的通信信道&#xff0c;支持使用Let’s Encrypt自动生成证书。Ligolo-ng的通信隐蔽性体现在其支持多种连接方式&#xff0c;适应复杂网…

铭豹扩展坞 USB转网口 突然无法识别解决方法

当 USB 转网口扩展坞在一台笔记本上无法识别,但在其他电脑上正常工作时,问题通常出在笔记本自身或其与扩展坞的兼容性上。以下是系统化的定位思路和排查步骤,帮助你快速找到故障原因: 背景: 一个M-pard(铭豹)扩展坞的网卡突然无法识别了,扩展出来的三个USB接口正常。…

未来机器人的大脑:如何用神经网络模拟器实现更智能的决策?

编辑&#xff1a;陈萍萍的公主一点人工一点智能 未来机器人的大脑&#xff1a;如何用神经网络模拟器实现更智能的决策&#xff1f;RWM通过双自回归机制有效解决了复合误差、部分可观测性和随机动力学等关键挑战&#xff0c;在不依赖领域特定归纳偏见的条件下实现了卓越的预测准…

Linux应用开发之网络套接字编程(实例篇)

服务端与客户端单连接 服务端代码 #include <sys/socket.h> #include <sys/types.h> #include <netinet/in.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <arpa/inet.h> #include <pthread.h> …

华为云AI开发平台ModelArts

华为云ModelArts&#xff1a;重塑AI开发流程的“智能引擎”与“创新加速器”&#xff01; 在人工智能浪潮席卷全球的2025年&#xff0c;企业拥抱AI的意愿空前高涨&#xff0c;但技术门槛高、流程复杂、资源投入巨大的现实&#xff0c;却让许多创新构想止步于实验室。数据科学家…

深度学习在微纳光子学中的应用

深度学习在微纳光子学中的主要应用方向 深度学习与微纳光子学的结合主要集中在以下几个方向&#xff1a; 逆向设计 通过神经网络快速预测微纳结构的光学响应&#xff0c;替代传统耗时的数值模拟方法。例如设计超表面、光子晶体等结构。 特征提取与优化 从复杂的光学数据中自…