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)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)