快捷導(dǎo)航

python opencv實(shí)現(xiàn)圖片旋轉(zhuǎn)矩形分割

更新時(shí)間：2018年07月26日 09:33:26 作者：Imcy

這篇文章主要為大家詳細(xì)介紹了python opencv實(shí)現(xiàn)圖片旋轉(zhuǎn)矩形分割，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下

有時(shí)候需要對(duì)有角度的矩形框內(nèi)圖像從原圖片中分割出來(lái)。這里的程序思想是，先將圖片進(jìn)行矩形角度的旋轉(zhuǎn)，使有角度的矩形處于水平狀態(tài)后，根據(jù)原來(lái)坐標(biāo)分割圖片。
參考：python opencv實(shí)現(xiàn)旋轉(zhuǎn)矩形框裁減功能

修改原來(lái)的程序：

1.旋轉(zhuǎn)函數(shù)的輸入僅為矩形的四點(diǎn)坐標(biāo)
2.角度由公式計(jì)算出來(lái)
3.矩形四點(diǎn)pt1,pt2,pt3,pt4由txt文件讀入
4.在旋轉(zhuǎn)程序中還處理了順時(shí)針和逆時(shí)針及出現(xiàn)矩形框翻轉(zhuǎn)的問(wèn)題。

代碼：

# -*- coding:utf-8 -*-
import cv2
from math import *
import numpy as np
import time,math
import os
import re

'''旋轉(zhuǎn)圖像并剪裁'''
def rotate(
    img, # 圖片
    pt1, pt2, pt3, pt4
):
  print pt1,pt2,pt3,pt4
  withRect = math.sqrt((pt4[0] - pt1[0]) ** 2 + (pt4[1] - pt1[1]) ** 2) # 矩形框的寬度
  heightRect = math.sqrt((pt1[0] - pt2[0]) ** 2 + (pt1[1] - pt2[1]) **2)
  print withRect,heightRect
  angle = acos((pt4[0] - pt1[0]) / withRect) * (180 / math.pi) # 矩形框旋轉(zhuǎn)角度
  print angle

  if pt4[1]>pt1[1]:
    print "順時(shí)針旋轉(zhuǎn)"
  else:
    print "逆時(shí)針旋轉(zhuǎn)"
    angle=-angle

  height = img.shape[0] # 原始圖像高度
  width = img.shape[1]  # 原始圖像寬度
  rotateMat = cv2.getRotationMatrix2D((width / 2, height / 2), angle, 1) # 按angle角度旋轉(zhuǎn)圖像
  heightNew = int(width * fabs(sin(radians(angle))) + height * fabs(cos(radians(angle))))
  widthNew = int(height * fabs(sin(radians(angle))) + width * fabs(cos(radians(angle))))

  rotateMat[0, 2] += (widthNew - width) / 2
  rotateMat[1, 2] += (heightNew - height) / 2
  imgRotation = cv2.warpAffine(img, rotateMat, (widthNew, heightNew), borderValue=(255, 255, 255))
  cv2.imshow('rotateImg2', imgRotation)
  cv2.waitKey(0)

  # 旋轉(zhuǎn)后圖像的四點(diǎn)坐標(biāo)
  [[pt1[0]], [pt1[1]]] = np.dot(rotateMat, np.array([[pt1[0]], [pt1[1]], [1]]))
  [[pt3[0]], [pt3[1]]] = np.dot(rotateMat, np.array([[pt3[0]], [pt3[1]], [1]]))
  [[pt2[0]], [pt2[1]]] = np.dot(rotateMat, np.array([[pt2[0]], [pt2[1]], [1]]))
  [[pt4[0]], [pt4[1]]] = np.dot(rotateMat, np.array([[pt4[0]], [pt4[1]], [1]]))

  # 處理反轉(zhuǎn)的情況
  if pt2[1]>pt4[1]:
    pt2[1],pt4[1]=pt4[1],pt2[1]
  if pt1[0]>pt3[0]:
    pt1[0],pt3[0]=pt3[0],pt1[0]

  imgOut = imgRotation[int(pt2[1]):int(pt4[1]), int(pt1[0]):int(pt3[0])]
  cv2.imshow("imgOut", imgOut) # 裁減得到的旋轉(zhuǎn)矩形框
  cv2.waitKey(0)
  return imgRotation # rotated image


?！「鶕?jù)四點(diǎn)畫(huà)原矩形
def drawRect(img,pt1,pt2,pt3,pt4,color,lineWidth):
  cv2.line(img, pt1, pt2, color, lineWidth)
  cv2.line(img, pt2, pt3, color, lineWidth)
  cv2.line(img, pt3, pt4, color, lineWidth)
  cv2.line(img, pt1, pt4, color, lineWidth)

#　讀出文件中的坐標(biāo)值
def ReadTxt(directory,imageName,last):
  fileTxt=directory+"http://rawLabel//"+imageName[:7]+last # txt文件名
  getTxt=open(fileTxt, 'r') # 打開(kāi)txt文件
  lines = getTxt.readlines()
  length=len(lines)
  for i in range(0,length,4):
    pt2=list(map(float,lines[i].split(' ')[:2]))
    pt1=list(map(float,lines[i+1].split(' ')[:2]))
    pt4=list(map(float,lines[i+2].split(' ')[:2]))
    pt3=list(map(float,re.split('\n| ',lines[i+3])[:2]))
    # float轉(zhuǎn)int

    pt2=list(map(int,pt2))
    pt1=list(map(int,pt1))
    pt4=list(map(int,pt4))
    pt3=list(map(int,pt3))

    imgSrc = cv2.imread(imageName)
    drawRect(imgSrc, tuple(pt1),tuple(pt2),tuple(pt3),tuple(pt4), (0, 0, 255), 2)
    cv2.imshow("img", imgSrc)
    cv2.waitKey(0)
    rotate(imgSrc,pt1,pt2,pt3,pt4)


if __name__=="__main__":
  directory = "G://grasp//grapCode//trainImage//jpg//4"
  last = 'cneg.txt'
  imageName="pcd0247r.png"
  ReadTxt(directory,imageName,last)

原帶角度的矩形框：