前言

在上一節(jié)我們通過(guò)使用NumPy的數(shù)組分割成功的在我們的圖像上畫(huà)了一個(gè)綠色的方塊，但是如果我們想畫(huà)一個(gè)單一的線條或者圓圈該怎么辦呢？NumPy沒(méi)有提供相關(guān)的功能，但是OpenCV提供了相關(guān)的函數(shù)，在本節(jié)就將為大家介紹三個(gè)基本的OpenCV畫(huà)圖方法：

cv2.line() #1cv2.rectangle() #2cv2.circle() #3

1 畫(huà)直線和矩形

在開(kāi)始我們用OpenCV畫(huà)我們的杰作之前，我們需要定義一個(gè)畫(huà)布

import numpy as np #1import cv2 #2canvas = np.zeros((300, 300, 3), dtype='uint8') #3

#1-2:

關(guān)于所需包的導(dǎo)入

#3：

我們使用np.zeros()方法構(gòu)造了一個(gè)300*300的NumPy數(shù)組，同時(shí)分配了三個(gè)顏色空間，分別表示Red,Green,Blue,正如zeros名字所描述的一樣，這個(gè)方法用0填充了這個(gè)數(shù)組的每一個(gè)元素。在np.zeros()的第二個(gè)變量是數(shù)據(jù)類型：dtype。由于我們需要用RGB格式來(lái)表示我們的圖像，它的取值范圍是[0,255]，所以我們用“uint8”就顯得至關(guān)重要了，如果不聲明的話np.zeros()默認(rèn)的變量類型是float64.

1.1 畫(huà)直線

green = (0, 255, 0) #4cv2.line(canvas, (0, 0), (300, 300), green) #5cv2.imshow('Canvas', canvas) #6cv2.waitKey(0) #7red = (0, 0, 255) #8cv2.line(canvas, (300, 0), (0, 300), red, 3) #9cv2.imshow('Canvas', canvas) #10cv2.waitKey(0) #11

#4:

我們定義了一個(gè)元組來(lái)表示綠色

#5-7:

我們?cè)赾anvas上畫(huà)了一條綠線從坐標(biāo)(0,0)到(300,300),將結(jié)果顯示出來(lái)，并等待按下任意按鍵

#8：

我們定義了一個(gè)元組來(lái)表示紅色，再次強(qiáng)調(diào)OpenCV是BGR模式而不是RGB模式

#9-11：

我們?cè)赾anvas上畫(huà)了一條紅線從坐標(biāo)(300,0)到(0,300),并且該線的線型為3個(gè)像素，這也是最后一個(gè)參數(shù)的含義，然后將結(jié)果顯示出來(lái)，并等待按下任意按鍵

1.2 畫(huà)矩形

cv2.rectangle(canvas, (10, 10), (60, 60), green) #12cv2.imshow('Canvas', canvas) #13cv2.waitKey(0) #14cv2.rectangle(canvas, (50, 200), (200, 225), red, 5) #15cv2.imshow('Canvas', canvas) #16cv2.waitKey(0) #17blue = (255, 0, 0) #18cv2.rectangle(canvas, (200, 50), (225, 125), blue, -1) #19cv2.imshow('Canvas', canvas) #20cv2.waitKey(0) #21

#12-14：

我們使用了cv2.rectangle()方法，這個(gè)方法與cv2.line()方法用法是一樣的，第一個(gè)參數(shù)表示我們想要在canvas這個(gè)畫(huà)布上進(jìn)行畫(huà)圖，第二個(gè)參數(shù)是我們矩形的開(kāi)始點(diǎn)(10,10)，第三個(gè)參數(shù)是我們矩形的結(jié)束點(diǎn)(60,60),通過(guò)這兩個(gè)點(diǎn)我們定義了一個(gè)50*50像素大小的區(qū)域，第四個(gè)參數(shù)是我們矩形邊框的顏色——綠色，然后將結(jié)果顯示出來(lái)，并等待按下任意按鍵。

#15-17：

在第15行代碼中，正如我們可以控制話直線的線型粗細(xì)程度，我們也可以控制畫(huà)矩形的線型粗細(xì)，在這行代碼中最后一個(gè)參數(shù)“5”，則表示我們將在canvas上畫(huà)出一個(gè)邊框粗細(xì)為5個(gè)像素大小的起點(diǎn)為(50, 200),終點(diǎn)為 (200, 225)的紅色矩形，然后將結(jié)果顯示出來(lái)，并等待按下任意按鍵。

#18-21

到目前為止，我們畫(huà)的都是圖形的邊框，如果我們想要填充這個(gè)邊框怎么辦呢？ 在第19行代碼中：

cv2.rectangle(canvas, (200, 50), (225, 125), blue, -1) #19

我們畫(huà)了一個(gè)起點(diǎn)為（200,50），終點(diǎn)為（225,125）的藍(lán)色矩形，當(dāng)我們將最后一個(gè)設(shè)置線型的參數(shù)設(shè)置為“-1”，我們便可以得到一個(gè)填充藍(lán)色的矩形。

2 畫(huà)圓形

畫(huà)圓形和畫(huà)直線和矩形是一樣容易的，但是它有有一些不同：

2.1 同心圓

canvas = np.zeros((300, 300, 3), dtype='uint8') #22(centerX, centerY) = (canvas.shape[1] // 2, canvas.shape[0] // 2) #23white = (255, 255, 255) #24for r in range(0, 175, 25): #25 cv2.circle(canvas, (centerX, centerY), r, white) #26cv2.imshow('Canvas', canvas) #27cv2.waitKey(0) #28for i in range(0, 25): #29 radius = np.random.randint(5, high=200) #30 color = np.random.randint(0, high=256, size=(3,)) #31 pt = np.random.randint(0, high=300, size=(2,)) #32 cv2.circle(canvas, tuple(pt), radius, color, 1) #33cv2.imshow('Canvas', canvas) #34cv2.waitKey(0) #35

#22：

我們重新將我們的畫(huà)板變成一個(gè)白板

#23：

我們定義并計(jì)算一個(gè)中心點(diǎn)的坐標(biāo)（centerX,centerY）,在第1節(jié)和第2節(jié)我們均提到過(guò)shape[0]表示圖片的高度，shape[1]表示圖片的寬度，分別取它們的一半，得到centerY,centerX

#24：

定義一個(gè)白色的像素

#25-28：

我們從[0,175)的范圍以跨度為25進(jìn)行循環(huán)來(lái)取圓的半徑，

cv2.circle(canvas, (centerX, centerY), r, white) #26

然后在第26行通過(guò)cv2.circle()來(lái)進(jìn)行畫(huà)圓，第一個(gè)參數(shù)表示在canvas上進(jìn)行繪畫(huà)，第二個(gè)參數(shù)表示圓心，第三個(gè)參數(shù)表示半徑，第四個(gè)參數(shù)表示顏色。然后將結(jié)果顯示出來(lái)，并等待按下任意按鍵。

2.2 隨機(jī)圓

讓我們來(lái)讓圓多一點(diǎn)趣味：

for i in range(0, 25): #29 radius = np.random.randint(5, high=200) #30 color = np.random.randint(0, high=256, size=(3,)) #31 pt = np.random.randint(0, high=300, size=(2,)) #32 cv2.circle(canvas, tuple(pt), radius, color, -1) #33cv2.imshow('Canvas', canvas) #34cv2.waitKey(0) #35

#29：

通過(guò)循環(huán)表示，我們將畫(huà)25個(gè)圓

#30-32：

如果要畫(huà)圓，我們需要三個(gè)基本元素：圓的半徑，圓的顏色，圓的圓心。在這里我們需要通過(guò)np.random.randint使用NumPy中產(chǎn)生隨機(jī)數(shù)的能力。

radius = np.random.randint(5, high=200) #30

在第30行我們?cè)赱5,200)范圍內(nèi)產(chǎn)生隨機(jī)數(shù)

color = np.random.randint(0, high=256, size=(3,)) #31

在第31行我們將在[0,256)范圍內(nèi)產(chǎn)生隨機(jī)數(shù)，由于顏色值擁有3個(gè)通道值，我們通過(guò)size=(3,)讓NumPy返回一個(gè)包含三個(gè)元素的列表。

pt = np.random.randint(0, high=300, size=(2,)) #32

在第32行我們將在[0,300)范圍內(nèi)產(chǎn)生隨機(jī)數(shù)，我們通過(guò)size=(2,)讓NumPy返回一個(gè)包含2個(gè)元素的列表來(lái)表示圓心。

#33：

cv2.circle(canvas, tuple(pt), radius, color, -1) #33

在第33行中，我們通過(guò)調(diào)用cv2.circle()方法來(lái)畫(huà)圓，第一個(gè)參數(shù)表示在canvas上畫(huà)圓，第二個(gè)參數(shù)圓心坐標(biāo)需要注意必須是元組，所以使用tuple來(lái)進(jìn)行強(qiáng)制轉(zhuǎn)化列表，第三個(gè)參數(shù)是半徑，第四個(gè)參數(shù)表示顏色，第5個(gè)參數(shù)為“-1”，表示產(chǎn)生實(shí)心圓。

#34-35： 將結(jié)果顯示出來(lái)，并等待按下任意按鍵。

3 完整代碼

新建drawing.py

import numpy as npimport cv2canvas = np.zeros((300, 300, 3), dtype='uint8')green = (0, 255, 0)cv2.line(canvas, (0, 0), (300, 300), green)cv2.imshow('Canvas', canvas)cv2.waitKey(0)red = (0, 0, 255)cv2.line(canvas, (300, 0), (0, 300), red, 3)cv2.imshow('Canvas', canvas)cv2.waitKey(0)cv2.rectangle(canvas, (10, 10), (60, 60), green)cv2.imshow('Canvas', canvas)cv2.waitKey(0)cv2.rectangle(canvas, (50, 200), (200, 225), red, 5)cv2.imshow('Canvas', canvas)cv2.waitKey(0)blue = (255, 0, 0)cv2.rectangle(canvas, (200, 50), (225, 125), blue, -1)cv2.imshow('Canvas', canvas)cv2.waitKey(0)canvas = np.zeros((300, 300, 3), dtype='uint8')(centerX, centerY) = (canvas.shape[1] // 2, canvas.shape[0] // 2)white = (255, 255, 255)for r in range(0, 175, 25): cv2.circle(canvas, (centerX, centerY), r, white)cv2.imshow('Canvas', canvas)cv2.waitKey(0)for i in range(0, 25): radius = np.random.randint(5, high=200) color = np.random.randint(0, high=256, size=(3,)) pt = np.random.randint(0, high=300, size=(2,)) cv2.circle(canvas, tuple(pt), radius, color, -1)cv2.imshow('Canvas', canvas)cv2.waitKey(0)

4 效果展示

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持好吧啦網(wǎng)。