如何使用Python+OpenCV+Keras實(shí)現(xiàn)無(wú)口罩車輛駕駛員懲罰生成
介紹
我們知道,在當(dāng)前形勢(shì)下,我們正在逐步穩(wěn)定地克服大流行病的情況,而且情況每天都在改善。但是,眾所周知,即使已經(jīng)開始接種疫苗,徹底根除該病毒仍需花費(fèi)很多年。因此,為安全起見(jiàn),在接下來(lái)的幾年中,我們所有人都可能會(huì)習(xí)慣于戴口罩。就違反交通規(guī)則而言,政府仍然嚴(yán)格對(duì)在路上開車不戴口罩的人處以罰款。建立一個(gè)系統(tǒng),能夠追蹤所有交通違法者的細(xì)節(jié),提高公民的意識(shí)和紀(jì)律,是一個(gè)非常有用的辦法。還可以通過(guò)另外創(chuàng)建一個(gè)儀表板監(jiān)視程序來(lái)跟蹤該交通規(guī)則違反者的增加或減少,在給定期間內(nèi)收集罰款,確定主要違反規(guī)則的人群,從而改善該系統(tǒng)。工作范圍作為代碼實(shí)現(xiàn)的一部分,我們計(jì)劃設(shè)計(jì)一個(gè)模型,將圖像分類為戴口罩的和沒(méi)戴口罩的。對(duì)于獲得的屬于沒(méi)戴口罩類別的圖像,我們獲取車輛號(hào)牌的圖像,并嘗試提取車輛詳細(xì)信息。車牌識(shí)別使用第二個(gè)模型完成,該模型接收帶有車牌的輸入作為汽車圖像。一旦完成了車輛ID,我們便將詳細(xì)信息傳遞到虛擬數(shù)據(jù)庫(kù),該數(shù)據(jù)庫(kù)包含車牌持有人的數(shù)據(jù)以及車輛的詳細(xì)信息。根據(jù)數(shù)據(jù)驗(yàn)證,我們將產(chǎn)生罰款,該罰款將直接發(fā)送到違規(guī)者的家庭住址。軟件架構(gòu)
網(wǎng)頁(yè)抓取圖片
項(xiàng)目始于確定要使用的數(shù)據(jù)集的問(wèn)題。在我們的項(xiàng)目中,在線沖浪幾乎無(wú)法為我們提供可用于我們項(xiàng)目的現(xiàn)有數(shù)據(jù)集。因此,我們決定應(yīng)用網(wǎng)絡(luò)抓取收集戴口罩和沒(méi)戴口罩的圖像。我們使用Beautiful Soap and Requests庫(kù)從網(wǎng)站下載圖像并將其保存到包含戴口罩和沒(méi)帶口罩駕駛員的不同文件夾中。我們從下面的URL中提取了數(shù)據(jù),這些URL由已屏蔽圖像和未屏蔽圖像組成。鏈接url1 = https://www.gettyimages.in/photos/driving-mask?page=鏈接url2 = https://www.gettyimages.in/photos/driving-without-mask?page=下面是一段代碼,演示了Web上的圖像抓取。from bs4 import *
import requests as rq
import os
url1 = 'https://www.gettyimages.in/photos/driving-mask?page='
url2 = '&phrase=driving%20mask&sort=mostpopular'
url_list=[]
Links = []
for i in range(1,56):
full_url = url1+str(i)+url2
url_list.a(chǎn)ppend(full_url)
for lst in url_list:
r2 = rq.get(lst)
soup = BeautifulSoup(r2.text, 'html.parser')
x=soup.select('img[src^="https://media.gettyimages.com/photos/"]')
for img in x:
Links.a(chǎn)ppend(img['src'])
print(len(Links))
for index, img_link in enumerate(Links):
if i <= len(Links):
img_data = rq.get(img_link).content
with open("Masked_Drivers/" + str(index + 1) + '.jpg', 'wb+') as f:
f.write(img_data)
i += 1
圖像預(yù)處理在將圖像發(fā)送到模型之前,我們需要應(yīng)用一些清理技術(shù),例如圖像大小調(diào)整圖像的灰度和將像素重新縮放為較低的值。之后,圖像和目標(biāo)將保存在陣列中。import cv2,os
data_path='Dataset'
categories=os.listdir(data_path)
labels=[i for i in range(len(categories))]
label_dict=dict(zip(categories,labels)) #empty dictionary
print(label_dict)
print(categories)
print(labels)
img_size=100
data=[]
target=[]
img_size=100
data=[]
target=[]
for category in categories:
folder_path=os.path.join(data_path,category)
img_names=os.listdir(folder_path)
for img_name in img_names:
img_path=os.path.join(folder_path,img_name)
img=cv2.imread(img_path)
try:
gray=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
#Coverting the image into gray scale
resized=cv2.resize(gray,(img_size,img_size))
#resizing the gray scale into 50x50, since we need a fixed common size for all the images in the dataset
data.a(chǎn)ppend(resized)
target.a(chǎn)ppend(label_dict[category])
#appending the image and the label(categorized) into the list (dataset)
except Exception as e:
print('Exception:',e)
import numpy as np
data=np.a(chǎn)rray(data)/255.0
data=np.reshape(data,(data.shape[0],img_size,img_size,1))
target=np.a(chǎn)rray(target)
from keras.utils import np_utils
new_target=np_utils.to_categorical(target)
np.save('data',data)
np.save('target',new_target)
建立模型在這里,我們使用Keras的Sequential設(shè)計(jì)CNN模型。使用大約200個(gè)神經(jīng)元作為輸入來(lái)構(gòu)建CNN。應(yīng)用激活函數(shù)和最大池化技術(shù)后,我們將獲得另一組輸出特征,這些特征將通過(guò)Conv2D的另一層傳遞。最后,我們從softmax中獲得2個(gè)輸出,它們代表輸入圖像的戴口罩或未戴口罩狀態(tài)。from keras.models import Sequential
from keras.layers import Dense,Activation,F(xiàn)latten,Dropout
from keras.layers import Conv2D,MaxPooling2D
from keras.callbacks import ModelCheckpoint
model=Sequential()
model.a(chǎn)dd(Conv2D(200,(3,3),input_shape=data.shape[1:]))
model.a(chǎn)dd(Activation('relu'))
model.a(chǎn)dd(MaxPooling2D(pool_size=(2,2)))
model.a(chǎn)dd(Conv2D(100,(3,3)))
model.a(chǎn)dd(Activation('relu'))
model.a(chǎn)dd(MaxPooling2D(pool_size=(2,2)))
model.a(chǎn)dd(Flatten())
model.a(chǎn)dd(Dropout(0.5))
model.a(chǎn)dd(Dense(50,activation='relu'))
model.a(chǎn)dd(Dense(2,activation='softmax'))
model.compile(loss='categorical_crossentropy',optimizer='adam',metrics=['accuracy'])
訓(xùn)練和測(cè)試模型構(gòu)建CNN模型后,我們將數(shù)據(jù)分為訓(xùn)練集和測(cè)試集。后來(lái),我們通過(guò)設(shè)置各種參數(shù)(例如時(shí)期,訓(xùn)練集,驗(yàn)證集和驗(yàn)證拆分值),在訓(xùn)練和測(cè)試數(shù)據(jù)上擬合CNN模型。from sklearn.model_selection import train_test_split
train_data,test_data,train_target,test_target=train_test_split(data,target,test_size=0.4)
checkpoint = ModelCheckpoint('model-{epoch:03d}.model',monitor='val_loss',verbose=0,save_best_only=True,mode='auto')
history=model.fit(train_data,train_target,epochs=20,callbacks=[checkpoint],validation_split=0.1)
發(fā)表評(píng)論
請(qǐng)輸入評(píng)論內(nèi)容...
請(qǐng)輸入評(píng)論/評(píng)論長(zhǎng)度6~500個(gè)字
最新活動(dòng)更多
-
即日-11.13立即報(bào)名>>> 【在線會(huì)議】多物理場(chǎng)仿真助跑新能源汽車
-
11月28日立即報(bào)名>>> 2024工程師系列—工業(yè)電子技術(shù)在線會(huì)議
-
12月19日立即報(bào)名>> 【線下會(huì)議】OFweek 2024(第九屆)物聯(lián)網(wǎng)產(chǎn)業(yè)大會(huì)
-
即日-12.26火熱報(bào)名中>> OFweek2024中國(guó)智造CIO在線峰會(huì)
-
即日-2025.8.1立即下載>> 《2024智能制造產(chǎn)業(yè)高端化、智能化、綠色化發(fā)展藍(lán)皮書》
-
精彩回顧立即查看>> 【限時(shí)免費(fèi)下載】TE暖通空調(diào)系統(tǒng)高效可靠的組件解決方案
推薦專題
- 1 【一周車話】沒(méi)有方向盤和踏板的車,你敢坐嗎?
- 2 特斯拉發(fā)布無(wú)人駕駛車,還未迎來(lái)“Chatgpt時(shí)刻”
- 3 特斯拉股價(jià)大跌15%:Robotaxi離落地還差一個(gè)蘿卜快跑
- 4 馬斯克給的“驚喜”夠嗎?
- 5 打完“價(jià)格戰(zhàn)”,大模型還要比什么?
- 6 馬斯克致敬“國(guó)產(chǎn)蘿卜”?
- 7 神經(jīng)網(wǎng)絡(luò),誰(shuí)是盈利最強(qiáng)企業(yè)?
- 8 比蘋果偉大100倍!真正改寫人類歷史的智能產(chǎn)品降臨
- 9 諾獎(jiǎng)進(jìn)入“AI時(shí)代”,人類何去何從?
- 10 Open AI融資后成萬(wàn)億獨(dú)角獸,AI人才之爭(zhēng)開啟
- 高級(jí)軟件工程師 廣東省/深圳市
- 自動(dòng)化高級(jí)工程師 廣東省/深圳市
- 光器件研發(fā)工程師 福建省/福州市
- 銷售總監(jiān)(光器件) 北京市/海淀區(qū)
- 激光器高級(jí)銷售經(jīng)理 上海市/虹口區(qū)
- 光器件物理工程師 北京市/海淀區(qū)
- 激光研發(fā)工程師 北京市/昌平區(qū)
- 技術(shù)專家 廣東省/江門市
- 封裝工程師 北京市/海淀區(qū)
- 結(jié)構(gòu)工程師 廣東省/深圳市