加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
validate_label.py 679 Bytes
一键复制 编辑 原始数据 按行查看 历史
HY-Xie 提交于 2023-07-14 12:17 . 新增yolo格式解析验证
import cv2
image_path = r"D:\MyCodes\yolo_format\dataset\images\pic2.jpg"
ann_path = r"D:\MyCodes\yolo_format\dataset\labels\pic2.txt"
img = cv2.imread(image_path)
h, w, c = img.shape
with open(ann_path, mode='r', encoding='utf-8') as f:
for line in f.readlines():
def split_elm(*l):
return tuple(map(float, l))
cls, cx, cy, sw, sh = split_elm(*(line.split(' ')))
x1 = int(w * (cx - 0.5 * sw))
x2 = int(w * (cx + 0.5 * sw))
y1 = int(h * (cy - 0.5 * sh))
y2 = int(h * (cy + 0.5 * sh))
cv2.rectangle(img, (x1, y1), (x2, y2), (255, 255, 0), 2)
cv2.imshow("Rect", img)
cv2.waitKey(0)
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化