欢迎您访问365答案网,请分享给你的朋友!
生活常识 学习资料

labelmeCOCO格式数据与左上角右下角格式数据相互转化

时间:2023-05-26

import cv2class DataConvert(): """ 表达方式说明 x1,y1,x2,y2(x1,y1)为左上角坐标,(x2,y2)为右下角坐标 x1,y1,w,h(x1,y1)为左上角坐标,w为目标区域宽度,h为目标区域高度 xc,yc,w,h(xc,yc)为目标区域中心坐标,w为目标区域宽度,h为目标区域高度 COCO标注 """ def __init__(self): pass @staticmethod def cvtx0y0whTox1y1x2y2(x0, y0, w, h, imgShape): # "0.530921 0.666667 0.622368 0.666667"=>(167, 169, 639, 507) # labelme 的COCO标注格式就是 中心点x+中心点y+宽+高 (归一化的) # 此函数出来的就是 左上点 右下点 (未归一化的) height, width, c = imgShape x1, y1, x2, y2 = int((x0 - w * 0.5) * width), int((y0 - h * 0.5) * height), int((x0 + w * 0.5) * width), int((y0 + h * 0.5) * height) return x1, y1, x2, y2 @staticmethod def cvtx1y1x2y2Tox0y0wh(x1, y1, x2, y2, imgShape): # (167, 169, 639, 507)=>"0.530921 0.666667 0.622368 0.666667" # 左上点 右下点 (未归一化的) => 中心点x+中心点y+宽+高 (归一化的) height, width, c = imgShape x0, y0, w, h = (x1 + x2) / 2 / width, (y1 + y2) / 2 / height, (x2 - x1) / width, (y2 - y1) / height, return x0, y0, w, hif __name__ == '__main__': img = cv2.imread("jg02387.jpg") shape = img.shape x1, y1, x2, y2 = DataConvert.cvtx0y0whTox1y1x2y2(0.530921, 0.666667, 0.622368, 0.666667, shape) print(x1, y1, x2, y2) print(DataConvert.cvtx1y1x2y2Tox0y0wh(x1, y1, x2, y2, shape))

Copyright © 2016-2020 www.365daan.com All Rights Reserved. 365答案网 版权所有 备案号:

部分内容来自互联网,版权归原作者所有,如有冒犯请联系我们,我们将在三个工作时内妥善处理。