yolov5输出检测框的中心位置，框的长宽，框的位置，以及输出对应标签格式的输出-Toy模板网

这篇具有很好参考价值的文章主要介绍了yolov5输出检测框的中心位置，框的长宽，框的位置，以及输出对应标签格式的输出。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

yolov5输出检测框的中心位置，框的长宽，框的位置，以及输出对应标签格式的输出。模型读取每个图片，并将上述的信息依此输出到同名的txt文件中保存，具体需要哪些坐标可以自己选。

更改信息在detect.py文件中，需要更改的第一个地方在如下地方，输出位置在runs/detect/exp中
yolov5detect检测框部分,python,yolo,检测,目标检测

            # 自己改的部分
            location_center_dir = str(save_dir) + '/detect_location'
            if not os.path.exists(location_center_dir):
                os.makedirs(location_center_dir)
            location_center_path = location_center_dir + '\\' + str(p.stem) + (
                '' if dataset.mode == 'image' else f'_{frame}')  #
            flocation = open(location_center_path + '.txt', 'a')

第二个需要更改的地方：
yolov5detect检测框部分,python,yolo,检测,目标检测


                    # 自己改的部分
                    x0 = (int(xyxy[0].item()) + int(xyxy[2].item())) / 2
                    y0 = (int(xyxy[1].item()) + int(xyxy[3].item())) / 2  # 中心点坐标(x0, y0)

                    chang = int(xyxy[2].item()) - int(xyxy[0].item())
                    kuan = int(xyxy[3].item()) - int(xyxy[1].item())
                    # class_index = cls  # 获取属性
                    # object_name = names[int(cls)]  # 获取标签名如：person
                    label = int(cls)  # 对应每个物体的标签对应的数字label，如person:0
                    x0 = format(x0/2448, '.6f')  #此处我要保存和训练标注的txt一样的格式，故保留六位小数
                    y0 = format(y0/2048, '.6f')
                    chang = format(chang/2448, '.6f') # 2448和2048为我的图片长和宽
                    kuan = format(kuan/2048, '.6f')
                    if label==0:  # 在这里我只需要保存person的信息，可以删去，也可以自己更改
                        flocation.write(str(label) + ' ' + str(x0) + ' ' + str(y0) + ' '+str(chang)+' ' +str(kuan)+'\n')