【杂七杂八的代码】【1】python标注自己的目标检测数据集——yolo（darknet）格式（一）

yjyjjyjy1231

于 2019-10-23 09:38:59 发布

阅读量880

点赞数

分类专栏：目标检测系列杂七杂八的代码文章标签：杂七杂八的代码目标检测 yolo

本文链接：https://blog.youkuaiyun.com/qq_39055818/article/details/102694862

版权

本文介绍了如何使用Python为单类别多目标的数据集创建YOLO（Darknet）格式的标注，通过键盘操作添加和保存边界框，简化了标注过程。程序会自动在文件夹下创建txt标注文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

2019.10.28更新，这篇是针对单帧单类别多目标的目标检测数据标注的。由于我自己的数据是一段段的连续视频帧，因此其实可以加入目标跟踪，实现半自动的标注，可以省不少的精力，代码放在【杂七杂八的代码】【三】python标注自己的目标检测数据集——yolo（darknet）格式（二）

最近需要标注一些自己的数据跑yolo，之前用的是labelimg + XmlToTxt，但因为自己的数据集是单类别的，所以这种方法稍显繁琐，就自己花了一点时间写了标注的代码：

import os
import cv2

global img
global point1, point2
global g_rect
global have_point1

def on_mouse(event, x, y, flags, param):
    global img, point1, point2, g_rect, have_point1
    img2 = img.copy()
    if event == cv2.EVENT_LBUTTONDOWN:
        # print("1-EVENT_LBUTTONDOWN")
        if not have_point1:
            point1 = (x, y)
            have_point1 = True
        else:
            point2 = (x, y)
            have_point1 = False
            cv2.rectangle(img2, point1, point2, (0, 255, 0), thickness=1)
            cv2.imshow('image', img2)

def get_image_roi(bgr_image)

最低0.47元/天解锁文章