快捷導(dǎo)航

python+opencv實(shí)現(xiàn)文字顏色識(shí)別與標(biāo)定功能

更新時(shí)間：2021年09月13日 10:47:39 作者：AI炮灰

最近小編接了一個(gè)比較簡單的圖像處理的單子，今天小編給大家分享python+opencv實(shí)現(xiàn)文字顏色識(shí)別與標(biāo)定功能的完整思路及代碼，感興趣的朋友一起看看吧

最近接了一個(gè)比較簡單的圖像處理的單子，花了一點(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)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: