Python OpenCV實現測量圖片物體寬度
一、 題目描述
測量所給圖片的高度,即上下邊緣間的距離。
思路:
將圖片進行閾值操作得到二值化圖片。 截取只包含上下邊框的部分,以便于后續的輪廓提取 輪廓檢測 得到結果二、 實現過程
1.用于給圖片添加中文字符
#用于給圖片添加中文字符def ImgText_CN(img, text, left, top, textColor=(0, 255, 0), textSize=20): if (isinstance(img, np.ndarray)): #判斷是否為OpenCV圖片類型 img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB)) draw = ImageDraw.Draw(img) fontText = ImageFont.truetype(r’C:WindowsFontssimsun.ttc’, textSize, encoding='utf-8') ##中文字體 draw.text((left, top), text, textColor, font=fontText) #寫文字 return cv2.cvtColor(np.asarray(img), cv2.COLOR_RGB2BGR)
2.實現圖片反色功能
#實現圖片反色功能def PointInvert(img): height, width = img.shape #獲取圖片尺寸 for i in range(height): for j in range(width): pi = img[i, j] img[i, j] = 255 - pi return img
3.邊緣檢測
# canny邊緣檢測edges = cv2.Canny(th, 30, 70) res=PointInvert(edges) #顏色反轉#顯示圖片cv2.imshow(’original’, th) #顯示二值化后的圖,主題為白色,背景為黑色 更加容易找出輪廓key = cv2.waitKey(0)if key==27: #按esc鍵時,關閉所有窗口 print(key) cv2.destroyAllWindows()
4.輪廓操作
contours, hierarchy = cv2.findContours(th, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE) #得到輪廓cnt = contours[0]#取出輪廓x, y, w, h = cv2.boundingRect(cnt) #用一個矩形將輪廓包圍img_gray = cv2.cvtColor(res, cv2.COLOR_GRAY2BGR)#將灰度轉化為彩色圖片方便畫圖cv2.line(img_gray, (x, y), (x + w, y), (0,0,255), 2, 5) #上邊緣cv2.line(img_gray, (x, y+h), (x + w, y+h), (0, 0, 255), 2, 5) #下邊緣img1[80:230, 90:230] = img_gray #用帶有上下輪廓的圖替換掉原圖的對應部分
5.顯示圖片
res1=ImgText_CN(img1, ’寬度%d’%h, 25, 25, textColor=(0, 255, 0), textSize=30) #繪制文字#顯示圖片 cv2.imshow(’original’, res1)key = cv2.waitKey(0)if key==27: #按esc鍵時,關閉所有窗口 print(key) cv2.destroyAllWindows()
6.完整代碼
import cv2import numpy as npfrom PIL import Image, ImageDraw, ImageFont#用于給圖片添加中文字符def ImgText_CN(img, text, left, top, textColor=(0, 255, 0), textSize=20): if (isinstance(img, np.ndarray)): #判斷是否為OpenCV圖片類型 img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB)) draw = ImageDraw.Draw(img) fontText = ImageFont.truetype(r’C:WindowsFontssimsun.ttc’, textSize, encoding='utf-8') ##中文字體 draw.text((left, top), text, textColor, font=fontText) #寫文字 return cv2.cvtColor(np.asarray(img), cv2.COLOR_RGB2BGR)#實現圖片反色功能def PointInvert(img): height, width = img.shape #獲取圖片尺寸 for i in range(height): for j in range(width): pi = img[i, j] img[i, j] = 255 - pi return imgimg=cv2.imread('gongjian1.bmp',0)#加載彩色圖img1=cv2.imread('gongjian1.bmp',1)#加載灰度圖recimg = img[80:230, 90:230] #截取需要的部分img2 = img1[80:230, 90:230] #截取需要的部分ret, th = cv2.threshold(recimg, 90, 255, cv2.THRESH_BINARY) #閾值操作二值化# canny邊緣檢測edges = cv2.Canny(th, 30, 70) res=PointInvert(edges) #顏色反轉#顯示圖片cv2.imshow(’original’, th) #顯示二值化后的圖,主題為白色,背景為黑色 更加容易找出輪廓key = cv2.waitKey(0)if key==27: #按esc鍵時,關閉所有窗口 print(key) cv2.destroyAllWindows() contours, hierarchy = cv2.findContours(th, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE) #得到輪廓cnt = contours[0]#取出輪廓x, y, w, h = cv2.boundingRect(cnt) #用一個矩形將輪廓包圍img_gray = cv2.cvtColor(res, cv2.COLOR_GRAY2BGR)#將灰度轉化為彩色圖片方便畫圖cv2.line(img_gray, (x, y), (x + w, y), (0,0,255), 2, 5) #上邊緣cv2.line(img_gray, (x, y+h), (x + w, y+h), (0, 0, 255), 2, 5) #下邊緣img1[80:230, 90:230] = img_gray #用帶有上下輪廓的圖替換掉原圖的對應部分res1=ImgText_CN(img1, ’寬度%d’%h, 25, 25, textColor=(0, 255, 0), textSize=30) #繪制文字#顯示圖片 cv2.imshow(’original’, res1)key = cv2.waitKey(0)if key==27: #按esc鍵時,關閉所有窗口 print(key) cv2.destroyAllWindows()
三、 運行結果(效果)
四、 問題及解決方法
紅色輪廓沒有顯示,解決方案:將灰度圖轉化為彩色圖
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持好吧啦網。
相關文章: