python ImageDraw類實現幾何圖形的繪制與文字的繪制
python PIL圖像處理模塊中的ImageDraw類支持各種幾何圖形的繪制和文本的繪制,如直線、橢圓、弧、弦、多邊形以及文字等。
下面直接通過示例來進行說明:
#-*- coding: UTF-8 -*- import numpy as npfrom PIL import Imagefrom PIL import ImageDrawfrom PIL import ImageFont def draw_test(): #生成深藍色繪圖畫布 array = np.ndarray((480, 640, 3), np.uint8) array[:, :, 0] = 0 array[:, :, 1] = 0 array[:, :, 2] = 100 image = Image.fromarray(array) #創建繪制對象 draw = ImageDraw.Draw(image) #繪制直線 draw.line((20, 20, 150, 150), ’cyan’) #繪制矩形 draw.rectangle((100, 200, 300, 400), ’black’, ’red’) #繪制弧 draw.arc((100, 200, 300, 400), 0, 180, ’yellow’) draw.arc((100, 200, 300, 400), -90, 0, ’green’) #繪制弦 draw.chord((350, 50, 500, 200), 0, 120, ’khaki’, ’orange’) #繪制圓餅圖 draw.pieslice((350, 50, 500, 200), -150, -30, ’pink’, ’crimson’) #繪制橢圓 draw.ellipse((350, 300, 500, 400), ’yellowgreen’, ’wheat’) #外切矩形為正方形時橢圓即為圓 draw.ellipse((550, 50, 600, 100), ’seagreen’, ’skyblue’) #繪制多邊形 draw.polygon((150, 180, 200, 180, 250, 120, 230, 90, 130, 100), ’olive’, ’hotpink’) #繪制文本 font = ImageFont.truetype('consola.ttf', 40, encoding='unic')#設置字體 draw.text((100, 50), u’Hello World’, ’fuchsia’, font) image.show() return
首先,通過ImageDraw類創建一個繪制對象draw;
draw.line():直線的繪制,第一個參數指定的是直線的端點坐標,形式為(x0, y0, x1, y1),第二個參數指定直線的顏色;
draw.rectangle():矩形繪制,第一個參數指定矩形的對角線頂點(左上和右下),形式為(x0, y0, x1, y1),第二個指定填充顏色,第三個參數指定邊界顏色;
draw.arc():(橢)圓弧的繪制,第一個參數指定弧所在橢圓的外切矩形,第二、三兩個參數分別是弧的起始和終止角度, 第四個參數是填充顏色,第五個參數是線條顏色;
draw.chord():弦的繪制,和弧類似,只是將弧的起始和終止點通過直線連接起來;
draw.pieslice():圓餅圖的繪制,和弧與弦類似,只是分別將起始和終止點與所在(橢)圓中心相連;
draw.ellipse():橢圓的繪制,第一個參數指定橢圓的外切矩形, 第二、三兩個參數分別指定填充顏色和線條顏色,當外切矩形是正方形時,橢圓即為圓;
draw.polygon():繪制多邊形,第一個參數為多邊形的端點,形式為(x0, y0, x1, y1, x2, y2,……),第二、三兩個參數分別指定填充顏色和線條顏色;
draw.text():文字的繪制,第一個參數指定繪制的起始點(文本的左上角所在位置),第二個參數指定文本內容,第三個參數指定文本的顏色,第四個參數指定字體(通過ImageFont類來定義)。
繪制結果如下:
最后,補充一下python中所支持的顏色,如下圖所示:
另外,顏色也可以使用'#'加上6位16進制字符串表示如“#ff0000”,則和“red”等價,前兩位表示R通道的值,中間兩位表示G通道的值,最后兩位表示B通道的值。
PS:opencv+python 實現基本圖形的繪制及文本的添加
import cv2import numpy as npimport os class Drawing(object): ''' 使用opencv繪制圖形,支持直線,矩形,圓形,橢圓,多邊形以及被標注文字添加 ''' chart_list = [’line’, ’rectangle’, ’circle’, ’ellipse’, ’polylines’, ’puttext’] def __init__(self, src_img, dst_img, chart, dict_args): self.src_img = os.path.normpath(src_img) self.dst_img = os.path.normpath(dst_img) self.chart = chart self.dict_args = dict_args # 顏色不傳默認為紅色 self.color = dict_args[’color’] if dict_args.has_key(’color’) else (0,0,255) # 線條粗細不傳默認為 2 self.thickness = dict_args[’thickness’] if dict_args.has_key(’thickness’) else 2 def handle(self): # 導入圖片 self.src_img = cv2.imread(self.src_img) if self.chart not in self.chart_list: print ’must input your right parameter’ return if self.chart == ’line’: # 畫直線 self.start = self.dict_args[’start’] self.end = self.dict_args[’end’] self.draw_line() elif self.chart == ’rectangle’: # 畫矩形 self.top_left = self.dict_args[’top_left’] self.bottom_right = self.dict_args[’bottom_right’] self.draw_rectangle() elif self.chart == ’circle’: # 畫圓形 self.center = self.dict_args[’center’] self.radius = self.dict_args[’radius’] self.draw_circle() elif self.chart == ’ellipse’: # 畫橢圓 self.center = self.dict_args[’center’] self.axes = self.dict_args[’axes’] # 旋轉角度,起始角度,終止角度 可不傳參,使用默認值 self.angle = self.dict_args[’angle’] if self.dict_args.has_key(’angle’) else 0 self.startangle = self.dict_args[’startangle’] if self.dict_args.has_key(’startangle’) else 0 self.endangle = self.dict_args[’endangle’] if self.dict_args.has_key(’endangle’) else 360 self.draw_ellipse() elif self.chart == ’polylines’: # 畫多邊形 if not isinstance(self.dict_args[’points’], list): self.pts = list(self.dict_args[’points’]) self.pts = np.array(self.dict_args[’points’], np.int32) self.close = self.dict_args[’close’] if self.dict_args.has_key(’close’) else True self.draw_polylines() else: # 標注文本 self.text = self.dict_args[’text’] self.position = self.dict_args[’position’] # 字體,文字大小 可不傳參,使用默認值 self.font = self.dict_args[’font’] if self.dict_args.has_key(’font’) else cv2.FONT_HERSHEY_SIMPLEX self.size = self.dict_args[’size’] if self.dict_args.has_key(’size’) else 1 self.add_text() cv2.imwrite(self.dst_img, self.src_img) def draw_line(self): # 劃線 # 輸入參數分別為圖像,開始坐標,結束坐標,顏色數組,粗細 cv2.line(self.src_img, self.start, self.end, self.color, self.thickness) def draw_rectangle(self): # 畫矩形 # 輸入參數分別為圖像、左上角坐標、右下角坐標、顏色數組、粗細 cv2.rectangle(self.src_img, self.top_left, self.bottom_right, self.color, self.thickness) def draw_circle(self): # 畫圓形 # 輸入參數為圖像,圓心,半徑,線條顏色,粗細 cv2.circle(self.src_img, self.center, self.radius, self.color, self.thickness) def draw_ellipse(self): # 畫橢圓 # 輸入參數為圖像,中心,(長軸,短軸),旋轉角度,起始角度,終止角度,線條顏色,粗細 cv2.ellipse(self.src_img, self.center, self.axes, self.angle, self.startangle,self.endangle, self.color, self.thickness) def draw_polylines(self): # 畫多邊形 # 輸入參數為圖像,多邊形各個頂點坐標,是否連成封閉圖形,線的顏色,粗細 cv2.polylines(self.src_img, [self.pts], self.close, self.color, self.thickness) def add_text(self): # 標注文本 # 輸入參數為圖像、文本、位置、字體、大小、顏色數組、粗細 cv2.putText(self.src_img, self.text, self.position, self.font, self.size, self.color, self.thickness)
以上就是python ImageDraw類實現幾何圖形的繪制與文字的繪制的詳細內容,更多關于python 幾何圖形的繪制的資料請關注好吧啦網其它相關文章!
相關文章: