您好,登錄后才能下訂單哦!
本篇內(nèi)容主要講解“Python中怎么使用cv2庫實現(xiàn)繪制動態(tài)圓和矩形”,感興趣的朋友不妨來看看。本文介紹的方法操作簡單快捷,實用性強。下面就讓小編來帶大家學習“Python中怎么使用cv2庫實現(xiàn)繪制動態(tài)圓和矩形”吧!
代碼如下
import cv2 import numpy as np import copy class Rect(object): #創(chuàng)建一個類用于接收鼠標點擊產(chǎn)生的數(shù)據(jù)坐標點。 def __init__(self): self.t1 = (0,0) self.br = (0,0) self.r = 0 def regularize(self): pt1 = (min(self.t1[0],self.br[0]),min(self.t1[1],self.br[1])) br1 = (max(self.t1[0],self.br[0]),max(self.t1[1],self.br[1])) self.t1 = pt1 self.br = br1 class DrawRects(object): def __init__(self,img,color = (0,255,0),thickness = 3): self.img = img self.image_for_show = img.copy() ''' 用于覆蓋之前的圖片,然后在新的空白圖片上將以前繪制的圖片和目前最新的圖片 繪制的到新的圖片上。 ''' self.color = color self.thickness = thickness self.left_button_down = False #用于判斷左鍵是否按下。 self.current_rect = Rect() #用于接收最新繪制的圖形。 self.rects = [] #用于接收繪制的矩形數(shù)據(jù)。 self.circles = [] #用于接收繪制的圓數(shù)據(jù) @staticmethod def __clip(value,low,high): output = max(value,low) output = min(output,high) return output def shrink_point(self,x,y): height,width = self.image_for_show.shape[0:2] x_shrink = self.__clip(x,0,width) y_shrink = self.__clip(y,0,height) return (x_shrink,y_shrink) ''' 上述方法用于限制鼠標點擊產(chǎn)生的數(shù)據(jù)在圖片范圍內(nèi)。 ''' def reset_image(self): self.image_for_show = self.img.copy() def append(self): if draw_circle_now: self.rects.append(['c',copy.deepcopy(self.current_rect)]) else: self.rects.append(['r',copy.deepcopy(self.current_rect)]) def draw(self): for rect in self.rects: if rect[0] == 'r': cv2.rectangle(self.image_for_show,rect[1].t1,rect[1].br,color = self.color,thickness=self.thickness) elif rect[0] == 'c': cv2.circle(self.image_for_show,rect[1].t1,rect[1].r,color = self.color,thickness = self.thickness) def cal_R(self): self.current_rect.r = int((abs(self.current_rect.br[0] - self.current_rect.t1[0])**2 + abs(self.current_rect.br[1] - self.current_rect.t1[1])**2)**0.5) #print(self.current_circle.r) return self.current_rect.r def draw_current_rect(self): if draw_circle_now: self.current_rect.r = self.cal_R() cv2.circle(self.image_for_show,self.current_rect.t1,self.current_rect.r,color = self.color,thickness=3) else: cv2.rectangle(self.image_for_show,self.current_rect.t1,self.current_rect.br, color = self.color,thickness = self.thickness) def pop(self): rect = Rect() if self.rects: rect = self.rects.pop() return rect def onmouse_draw_rect(event,x,y,flags,draw_rects): if event == cv2.EVENT_LBUTTONDOWN: draw_rects.left_button_down = True draw_rects.current_rect.t1 = (x,y) if draw_rects.left_button_down and event == cv2.EVENT_MOUSEMOVE: draw_rects.current_rect.br = draw_rects.shrink_point(x,y) draw_rects.reset_image() draw_rects.draw() draw_rects.draw_current_rect() if event == cv2.EVENT_LBUTTONUP: draw_rects.left_button_down = False draw_rects.current_rect.br = draw_rects.shrink_point(x,y) #draw_rects.current_rect.regularize() draw_rects.append() if (not draw_rects.left_button_down) and event == cv2.EVENT_RBUTTONDOWN: draw_rects.pop() draw_rects.reset_image() draw_rects.draw() draw_circle_now = False img = np.zeros((516,516,3),np.uint8) draw_rects = DrawRects(img,(0,255,255)) cv2.namedWindow('image') cv2.setMouseCallback('image',onmouse_draw_rect,draw_rects) while True: cv2.imshow('image',draw_rects.image_for_show) key = cv2.waitKey(30) if key == ord('c'): draw_circle_now = not draw_circle_now elif key == 27: break cv2.destroyAllWindows()
到此,相信大家對“Python中怎么使用cv2庫實現(xiàn)繪制動態(tài)圓和矩形”有了更深的了解,不妨來實際操作一番吧!這里是億速云網(wǎng)站,更多相關(guān)內(nèi)容可以進入相關(guān)頻道進行查詢,關(guān)注我們,繼續(xù)學習!
免責聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點不代表本網(wǎng)站立場,如果涉及侵權(quán)請聯(lián)系站長郵箱:is@yisu.com進行舉報,并提供相關(guān)證據(jù),一經(jīng)查實,將立刻刪除涉嫌侵權(quán)內(nèi)容。