旋转标注框数据标签制作-roLabelImg

 

一、roLabelImg简介

roLabelImg是基于labelImg改进的,也是用来标注为VOC格式的数据,但是在labelImg的基础上增加了能够使标注的框进行旋转的功能。

网址:https://github.com/cgvict/roLabelImg/blob/master/README.rst

 

二、windows上安装编译

详细可以参考labelImg里的说明(https://github.com/tzutalin/labelImg

1) 下载Anaconda(python3+)

2)打开命令窗口,输入如下命令

      $ conda install pyqt=5

      $ pyrcc5 -o libs/resources.py resources.qrc

      $ python labelImg.py , 或

      $  python labelImg.py [IMAGE_PATH] [PRE-DEFINED CLASS FILE]

3) 编译打包,可以使用如下命令:

      $  /PATH/TO/pyinstaller.exe --hidden-import=xml ^

            --hidden-import=xml.etree ^

            --hidden-import=xml.etree.ElementTree ^

            --hidden-import=lxml.etree ^

            -D -F -n roLabelImg -c "../labelImg.py" -p ../libs -p ../

 

三、使用方法

软件快捷键如下:

1) w: 创建水平矩形目标框

2) e: 创建旋转矩形目标框

3) zxcv: 旋转目标框,键z和建x是逆时针旋转,键c和键v是顺时针旋转

旋转标注框数据标签制作-roLabelImg_第1张图片

使用roLabelImg标注得到的目标框的信息如下:

旋转标注框数据标签制作-roLabelImg_第2张图片

  • width是图像的宽,height是图像的高,depth是图像的通道数
  • cx, cy表示bndbox的中心点坐标(坐标系方向和一般的图像坐标系相同,左上角为原点,向右为x正方向,向下为y正方向);
  • h和w是标记目标的高和宽;
  • angle是旋转角度信息,水平bndbox,angle=0,,顺时针方向旋转,得到的角度值是一个弧度单位的正值,且旋转一周为pi,没有负值。

 

四、标签目标格式转换

4.1 转换成四点坐标:

def rotatePoint(xc,yc, xp,yp, theta):

      xoff = xp-xc;

      yoff = yp-yc;

      cosTheta = math.cos(theta)

      sinTheta = math.sin(theta)

      pResx = cosTheta * xoff + sinTheta * yoff

      pResy = - sinTheta * xoff + cosTheta * yoff

      # pRes = (xc + pResx, yc + pResy)

      return xc+pResx,yc+pResy

 

def addRotatedShape(cx,cy,w,h,angle):

      p0x,p0y = self.rotatePoint(cx,cy, cx - w/2, cy - h/2, -angle)

      p1x,p1y = self.rotatePoint(cx,cy, cx + w/2, cy - h/2, -angle)

      p2x,p2y = self.rotatePoint(cx,cy, cx + w/2, cy + h/2, -angle)

      p3x,p3y = self.rotatePoint(cx,cy, cx - w/2, cy + h/2, -angle)

      points = [(p0x, p0y), (p1x, p1y), (p2x, p2y), (p3x, p3y)]

      return points

详细代码参考: pascal_voc_io.py(https://github.com/cgvict/roLabelImg/blob/master/libs/pascal_voc_io.py

 

你可能感兴趣的:(旋转标注框数据标签制作-roLabelImg)