【自己制作数据集】制作标注并生成mask

最新推荐文章于 2025-03-25 21:42:44 发布

haohulala

最新推荐文章于 2025-03-25 21:42:44 发布

阅读量9.4k

点赞数 6

分类专栏：计算机视觉

本文链接：https://blog.youkuaiyun.com/haohulala/article/details/111386785

版权

在深度学习的时候，我们可能需要自己对图片进行标注，这篇博客记录自己对图片进行标注并且生成mask的过程

首先使用pip安装标注工具 labelme

pip install labelme

安装完成后直接在cmd中输入labelme就会打开软件

使用软件读取图片，添加标注，添加完成后直接保存json文件就行了。

下面介绍如何根据标注生成mask，我们主要使用opencv进行多边形填充来生成mask，详细代码如下

import json
import cv2
import matplotlib.pyplot as plt
import numpy as np


tmp = {}
with open("./tmp.json", "r") as f:
    tmp = f.read()

tmp = json.loads(tmp)

points = tmp["shapes"][0]["points"]
points = np.array(points, np.int32)

img = cv2.imread("./tmp.jpg")
#BGR->RGB
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)

box = tmp["shapes"][1]["points"]
box = np.array(box, np.int32)

m