LabelImg标注的YOLO格式txt标签中心坐标和物体边界框长宽的转换

目录

1.LabelImg标注的YOLO格式的TXT标签

Opencv+YOLO-V3实现目标跟踪

YOLO-V3实时检测实现（opencv+python实现）——改进——＞更加的易懂

YOLO-V3实时检测实现（opencv+python实现）

1.LabelImg标注的YOLO格式的TXT标签

关于LabelImg下载及使用：标注工具 labelImg 的下载安装及使用

首先标注一张图片：

查看标签.txt文件：

提示：如果我们要进行训练的话，那么需要上面的坐标进行变换一下，得到框的左上角的坐标和右下角的坐标。

  

现在我们要根据中心坐标和边框得到左上角（xmin,ymin）和右下角（xmax,ymax）坐标：

变换公式如下：

提示：现在已经知道坐标之间的关系了，那么可以使用python进行求解：

• 第一步从标注的.txt文件中读取数据；
• 第二步将读取的数据利用上面的公式进行转换；

  

   

    

     

    
    

     

      def 
      Xmin_Xmax_Ymin_Ymax():
     
    
   

    

     

    
    

     

          img_path = 
      "data/train/000_0.png"
     
    
   

    

     

    
    

     

          txt_path = 
      "data/XML/000_0.txt"
     
    
   

    

     

    
    

     

          img = cv2.
      imread(img_path)
     
    
   

    

     

    
    

     

          # 获取图片的高宽
     
    
   

    

     

    
    

     

          h, w, _ = img.shape
     
    
   

    

     

    
    

     

          txt_path=
      "data/XML/000_0.txt"
     
    
   

    

     

    
    

     

          #读取TXT文件 中的中心坐标和框大小
     
    
   

    

     

    
    

     

          with 
      open(txt_path,
      "r") as fp:
     
    
   

    

     

    
    

     

              #以空格划分
     
    
   

    

     

    
    

     

              contline=fp.
      readline().
      split(
      ' ')
     
    
   

    

     

    
    

     

              #contline : class  x_center y_center width height
     
    
   

    

     

    
    

     
        
      print(contline)
     
    
   

    

     

    
    

     

          #计算框的左上角坐标和右下角坐标,使用strip将首尾空格去掉
     
    
   

    

     

    
    

     

          xmin=
      float((contline[
      1]).
      strip())
      -float(contline[
      3].
      strip())/
      2
     
    
   

    

     

    
    

     

          xmax=
      float(contline[
      1].
      strip())+
      float(contline[
      3].
      strip())/
      2
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          ymin = 
      float(contline[
      2].
      strip()) - 
      float(contline[
      4].
      strip()) / 
      2
     
    
   

    

     

    
    

     

          ymax = 
      float(contline[
      2].
      strip()) + 
      float(contline[
      4].
      strip()) / 
      2
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          #将坐标（
      0-
      1之间的值）还原回在图片中实际的坐标位置
     
    
   

    

     

    
    

     

          xmin,xmax=w*xmin,w*xmax
     
    
   

    

     

    
    

     

          ymin,ymax=h*ymin,h*ymax
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          return (xmin,xmax,ymin,ymax)
     
    
  
 

  
 

将返回的坐标利用opencv将框绘制出来：

  

   

    

     

    
    

     

      def 
      draw(tupelist):
     
    
   

    

     

    
    

     

          img_path = 
      "data/train/000_0.png"
     
    
   

    

     

    
    

     

          img = cv2.
      imread(img_path)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          xmin=tupelist[
      0]
     
    
   

    

     

    
    

     

          xmax=tupelist[
      1]
     
    
   

    

     

    
    

     

          ymin=tupelist[
      2]
     
    
   

    

     

    
    

     

          ymax=tupelist[
      3]
     
    
   

    

     

    
    

     

          cv2.
      rectangle(img,(
      int(xmin),
      int(ymin)),(
      int(xmax),
      int(ymax)),(
      255,
      0,
      255),
      2)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

          cv2.
      imshow(
      'img',img)
     
    
   

    

     

    
    

     

          cv2.
      waitKey(
      0)
     
    
   

    

     

    
    

     

          cv2.
      destroyAllWindows()
     
    
  

  

   

    

     

    
    

     

      import cv2
     
    
   

    

     

    
    

     

      import os
     
    
   

    

     

    
    

     

      import numpy 
      as np
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      if __name__ == '__main__':
     
    
   

    

     

    
    

     

          tuplelist=
      Xmin_Xmax_Ymin_Ymax()
     
    
   

    

     

    
    

     

          draw(tuplelist)