python利用四个坐标点对图片目标区域最小外接矩形进行裁剪

在图像裁剪操作中,opencv和pillow两个库都具有相应的函数,但是这两个库中的函数仅仅能对与图片平行的矩形进行裁剪操作,如果想要对目标的最小外接矩形进行裁剪该如何操作呢?如下所示:

具体处理该问题的思路如下:

  1. 计算最小外接矩形的四个点的坐标,旋转角度
  2. 将原图像进行旋转,旋转角度为最小外接矩形的角度
  3. 将四个点的坐标进行映射,求出被旋转后图像的四个点的坐标
  4. 利用这四个点对图像进行裁剪

图像原图如下:

1 求出该区域的最小外接矩形,并且得到外接矩形的四个点的坐标和旋转角度。

rect = cv2.minAreaRect(self.contours[0])#rect为[(旋转中心x坐标,旋转中心y坐标),(矩形长,矩形宽),旋转角度]
box_origin = cv2.boxPoints(rect)#box_origin为[(x0,y0),(x1,y1),(x2,y2),(x3,y3)]

2 将原图像进行旋转,旋转角度为最小外接矩形的角度,由于防止旋转后目标区域在图像外面,所以我将图像大小扩大为原来的2倍。

M = cv2.getRotationMatrix2D(rect[0],rect[2],1)
dst = cv2.warpAffine(rotateimg,M,(2*rotateimg.shape[0],2*rotateimg.shape[1]))

3 将原四个点的坐标做映射,映射到旋转后的区域,得到新的四个点的坐标。

#逆时针旋转
def Nrotate(angle,valuex,valuey,pointx,pointy):
      angle = (angle/180)*math.pi
      valuex = np.array(valuex)
      valuey = np.array(valuey)
      nRotatex = (valuex-pointx)*math.cos(angle) - (valuey-pointy)*math.sin(angle) + pointx
      nRotatey = (valuex-pointx)*math.sin(angle) + (valuey-pointy)*math.cos(angle) + pointy
      return (nRotatex, nRotatey)
#顺时针旋转
def Srotate(angle,valuex,valuey,pointx,pointy):
      angle = (angle/180)*math.pi
      valuex = np.array(valuex)
      valuey = np.array(valuey)
      sRotatex = (valuex-pointx)*math.cos(angle) + (valuey-pointy)*math.sin(angle) + pointx
      sRotatey = (valuey-pointy)*math.cos(angle) - (valuex-pointx)*math.sin(angle) + pointy
      return (sRotatex,sRotatey)
#将四个点做映射
def rotatecordiate(angle,rectboxs,pointx,pointy):
      output = []
      for rectbox in rectboxs:
        if angle>0:
          output.append(Srotate(angle,rectbox[0],rectbox[1],pointx,pointy))
        else:
          output.append(Nrotate(-angle,rectbox[0],rectbox[1],pointx,pointy))
      return output
box = rotatecordiate(rect[2],box_origin,rect[0][0],rect[0][1])

4 利用四个点坐标进行裁剪,如2中图所示,图像经过旋转后已经变为和图片没有旋转角的图像,经过实验观察旋转后的坐标还是有细微差别,但误差已经在千分点甚至万分点左右,对我们裁剪造成的影响可以忽略不计。

def imagecrop(image,box):
      xs = [x[1] for x in box]
      ys = [x[0] for x in box]
      print(xs)
      print(min(xs),max(xs),min(ys),max(ys))
      cropimage = image[min(xs):max(xs),min(ys):max(ys)]
      print(cropimage.shape)
      cv2.imwrite('cropimage.png',cropimage)
      return cropimage
imagecrop(dst,np.int0(box))

到此,利用4个坐标点对图像进行裁剪操作已经完成。

Python的GDAL库提供了裁剪栅格影像的功能,并且可以根据需要通过最小外接矩形裁剪影像。下面是用Python GDAL进行最小外接矩形裁剪的步骤: 首先,导入需要的库,包括GDAL库和OGR库。GDAL库用于处理栅格数据,OGR库用于处理矢量数据。 ```python from osgeo import gdal, ogr ``` 然后,打开需要裁剪的栅格影像,并获取其投影信息和地理转换信息。 ```python # 打开栅格影像 ds = gdal.Open("input.tif") # 获取投影信息和地理转换信息 proj = ds.GetProjection() gt = ds.GetGeoTransform() ``` 接下来,定义最小外接矩形的位置和大小。可以根据需要手动指定,也可以通过矢量数据计算得到。 ```python # 手动指定最小外接矩形位置和大小 xmin, ymin, xmax, ymax = 10, 20, 30, 40 ``` 然后,创建最小外接矩形的矢量数据。 ```python # 创建最小外接矩形的矢量数据 ring = ogr.Geometry(ogr.wkbLinearRing) ring.AddPoint(xmin, ymin) ring.AddPoint(xmin, ymax) ring.AddPoint(xmax, ymax) ring.AddPoint(xmax, ymin) ring.AddPoint(xmin, ymin) poly = ogr.Geometry(ogr.wkbPolygon) poly.AddGeometry(ring) ``` 接下来,设置裁剪的参数,并执行裁剪操作。 ```python # 设置裁剪参数 options = ["ALL_TOUCHED=TRUE"] # 执行裁剪操作 gdal.Warp("output.tif", ds, format="GTiff", cutlineDSName=poly, dstSRS=proj, outputType=gdal.GDT_Float32, options=options) ``` 最后,关闭数据集并输出裁剪后的影像。 ```python # 关闭数据集 ds = None # 输出裁剪后的影像 output = gdal.Open("output.tif") ``` 以上是使用Python GDAL进行最小外接矩形裁剪的步骤。通过指定最小外接矩形的位置和大小,可以裁剪出指定区域的栅格影像。
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值