python+opencv實(shí)現(xiàn)文字顏色識(shí)別與標(biāo)定功能
最近接了一個(gè)比較簡(jiǎn)單的圖像處理的單子,花了一點(diǎn)時(shí)間隨便寫了一下:
數(shù)據(jù)集客戶沒有是自己隨便創(chuàng)建的:
程序如下:
""" Code creation time:September 11, 2021 Author:PanBo Realize function:It mainly realizes the recognition and calibration of fonts with different colors """ import numpy as np import cv2 as cv font = cv.FONT_HERSHEY_SIMPLEX lower_red = np.array([0, 120, 120]) hight_red = np.array([10, 255, 255]) # lower_black = np.array([0, 0, 0]) height_black = np.array([144, 144, 144]) lower_yellow = np.array([10, 230, 230]) height_yellow = np.array([35, 255, 255]) frame = cv.imread("test.png") cv.namedWindow("test_image", cv.WINDOW_AUTOSIZE) cv.imshow('test_image', frame) img_hsv = cv.cvtColor(frame, cv.COLOR_BGR2HSV) mask_red = cv.inRange(img_hsv, lower_red, hight_red) mask_black = cv.inRange(img_hsv, lower_black, height_black) mask_yellow = cv.inRange(img_hsv, lower_yellow, height_yellow) cv.namedWindow("mask_red", cv.WINDOW_AUTOSIZE) cv.imshow("mask_red", mask_red) cv.namedWindow("mask_black", cv.WINDOW_AUTOSIZE) cv.imshow("mask_black", mask_black) cv.namedWindow("mask_yellow", cv.WINDOW_AUTOSIZE) cv.imshow("mask_yellow", mask_yellow) kernel = cv.getStructuringElement(cv.MORPH_RECT, (3, 3)) mask_yellow = cv.morphologyEx(mask_yellow, cv.MORPH_OPEN, kernel) cv.namedWindow("mask_yellow_open", cv.WINDOW_AUTOSIZE) cv.imshow("mask_yellow_open", mask_yellow) mask_yellow = cv.morphologyEx(mask_yellow, cv.MORPH_CLOSE, kernel) cv.namedWindow("mask_yellow_close", cv.WINDOW_AUTOSIZE) cv.imshow("mask_yellow_close", mask_yellow) mask_black = cv.medianBlur(mask_black, 3) mask_red = cv.medianBlur(mask_red, 3) mask_yellow = cv.medianBlur(mask_yellow, 3) # cv.imshow(" ", mask_green) cnts1, hierarchy1 = cv.findContours(mask_black, cv.RETR_EXTERNAL, cv.CHAIN_APPROX_NONE) cnts2, hierarchy2 = cv.findContours(mask_red, cv.RETR_EXTERNAL, cv.CHAIN_APPROX_NONE) cnts3, hierarchy3 = cv.findContours(mask_yellow, cv.RETR_EXTERNAL, cv.CHAIN_APPROX_NONE) for cnt in cnts1: (x, y, w, h) = cv.boundingRect(cnt) cv.rectangle(frame, (x, y), (x + w, y + h), (0, 0, 0), 2) cv.putText(frame, 'black', (x, y-5), font, 0.7, (0, 0, 25),2) for cnt in cnts2: (x, y, w, h) = cv.boundingRect(cnt) cv.rectangle(frame, (x, y), (x + w, y + h), (0, 0, 255), 2) cv.putText(frame, 'red', (x, y - 5), font, 0.7, (0, 0, 255), 2) for cnt in cnts3: (x, y, w, h) = cv.boundingRect(cnt) cv.rectangle(frame, (x, y), (x + w, y + h), (0, 0, 255), 2) cv.putText(frame, 'yellow', (x, y - 5), font, 0.7, (0, 0, 255), 2) cv.namedWindow("output", cv.WINDOW_AUTOSIZE) cv.imshow("output", frame) cv.waitKey(0)
首先給定字體顏色的上下閾值,然后提取出紅色、黑色以及黃色的字體如下所示:
檢測(cè)出原始圖像中存在的紅色字體的為四和五
檢車出原始圖像中存在的黑色字體是一和大
檢測(cè)出原始圖像中存在黃色字體的是七和九
但是通過yellow字體檢測(cè)的過程中發(fā)現(xiàn)有一些椒鹽噪聲需要去剔除,因此做了一下中值濾波
但是效果不好有做一個(gè)形態(tài)學(xué)操作——開操作,結(jié)果如下:
但是發(fā)現(xiàn)了七和九發(fā)生了斷層現(xiàn)象如果進(jìn)行矩形標(biāo)定的話會(huì)出現(xiàn)兩個(gè)矩形,因此又做了一下形態(tài)學(xué)操作中的——閉操作結(jié)果如下:
經(jīng)過閉操作我們發(fā)現(xiàn)七處的裂縫沒有了但是九還是有,這個(gè)是由于設(shè)置的yellow上下閾值導(dǎo)致的。
后面進(jìn)行矩形標(biāo)定如下所示:
到此這篇關(guān)于python+opencv實(shí)現(xiàn)文字顏色識(shí)別與標(biāo)定的文章就介紹到這了,更多相關(guān)python opencv文字顏色識(shí)別內(nèi)容請(qǐng)搜索本站以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持本站!
版權(quán)聲明:本站文章來源標(biāo)注為YINGSOO的內(nèi)容版權(quán)均為本站所有,歡迎引用、轉(zhuǎn)載,請(qǐng)保持原文完整并注明來源及原文鏈接。禁止復(fù)制或仿造本網(wǎng)站,禁止在非www.sddonglingsh.com所屬的服務(wù)器上建立鏡像,否則將依法追究法律責(zé)任。本站部分內(nèi)容來源于網(wǎng)友推薦、互聯(lián)網(wǎng)收集整理而來,僅供學(xué)習(xí)參考,不代表本站立場(chǎng),如有內(nèi)容涉嫌侵權(quán),請(qǐng)聯(lián)系alex-e#qq.com處理。