栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

将XYWH数据转换为YOLO数据格式

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

将XYWH数据转换为YOLO数据格式

1. YOLO数据格式介绍

YOLO标签的数据是一个相对值,而不是绝对值。

YOLO标签数据为原始图像对应的txt文件,每一张图像对应一个txt,其中包含了多条标签信息。

数据格式表示为:

1:[目标类别]

2:[目标中心点横坐标与图像宽度比值]

3:[目标中心点纵坐标与图像高度比值]

4:[目标框与图像宽度比值]

5:[目标框与图像高度比值] 

2. 代码

将bbox的绝对值转化为YOLO数据格式,此处以xywh为例:

import cv2

def xywh_to_yolo(img_path, x, y, w, h):
    # img_path: 原始图像路径
    img = cv2.imread(img_path)
    img_w, img_h = img.shape[1], img.shape[0]
    dw = 1./img_w
    dh = 1./img_h
    cx = (x + (x+w))/2.0
    cy = (y + (y+h))/2.0
    x = cx*dw
    y = cy*dh
    w = w*dw
    h = h*dh
    return (x, y, w, h)

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1030876.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号