从txt中读取数据转成yolo格式数据

import os
import cv2


def changeLable(path):
    fileList = os.listdir(path)
    for fileName in fileList:
        # 打开标签文件
        fileLabel = open(path + "/" + fileName, "r+")
        # 取出文件名并拼接.jpg
        imgName = fileName.split(".")[0]+".jpg"
        print("所需图片{}".format(imgName))
        imagePath = "E:/iron/coco/image/"+imgName
        print("图片地址{}".format(imagePath))
        img = cv2.imread(imagePath)
        h_img, w_img = img.shape[:2]
        print("图片宽:{}\t高:{}".format(w_img,h_img))
        info = fileLabel.read()
        kind = info.split(" ")[0]
        x1 = int(info.split(" ")[1])
        x2 = int(info.split(" ")[2])
        y2 = int(info.split(" ")[3])
        y1 = int(info.split(" ")[4])
        cenX = int(x1 + (x2-x1) / 2)
        cenY = int(y1 + (y2-y1) / 2)
        wide = (x2 - x1)
        high = (y2 - y1)
        print("中心点坐标({},{})\t宽:{}\t高:{}\n".format(cenX, cenY, wide, high))
        # 归一化
        newKind = "0"
        newcenX = cenX/w_img
        newcenY = cenY/h_img
        newWide = wide/w_img
        newHigh = high/h_img
        print("归一化后中心点坐标({},{})\t宽:{}\t高:{}\n".format(newcenX, newcenY, newWide, newHigh))
        info = newKind + " " + str(newcenX) + " " + str(newcenY) + " " + str(newWide) + " " + str(newHigh)
        print(info)
        newLabelPath="E:/iron/coco/label"
        file = open(newLabelPath + "/" + fileName, "w+")
        file.write(info)
        fileLabel.close()
        file.close()


if __name__ == "__main__":
    changeLable("E:/iron/original/label")