yolov5 数据集切分及格式转换

最新推荐文章于 2024-11-24 22:01:16 发布

我的膝盖中了一箭

最新推荐文章于 2024-11-24 22:01:16 发布

阅读量448

点赞数 6

文章标签： YOLO python 深度学习

本文链接：https://blog.youkuaiyun.com/weixin_38475691/article/details/140882563

版权

import xml.etree.ElementTree as ET
import os
import random

def vocbox2cocobox(bbox,size):
    xmin, ymin, xmax, ymax = bbox
    x = (xmin+xmax)/2. - 1
    y = (ymin+ymax)/2. - 1
    ix = x/size[0]
    iy = y/size[1]
    w = xmax-xmin
    h = ymax-ymin
    iw = w/size[0]
    ih = h/size[1]
    return ix,iy,iw,ih
    pass

def load_xml(path,save_path):
    cls2id = {}
    save_file =  open(save_path,'w')
    tree = ET.parse(open(path))
    root = tree.getroot()
    size = root.find('size')
    h = int(size.find('height').text)
    w = int(size.find('width').text)
    content = ''
    for obj in root.iter('object'):
        cls = obj.find('name').text
        # if cls not in cls2id:
        #     continue
        bbox = obj.find('bndbox')
        xmin = float(bbox.find('xmin').text)
        ymin