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

ゞ浴缸里的玫瑰 2023-09-28 16:13 5阅读 0赞

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")