本技術(shù)涉及電池,尤其涉及一種圖像檢測(cè)模型及其訓(xùn)練方法、電池檢測(cè)方法和設(shè)備。
背景技術(shù):
1、節(jié)能減排是汽車(chē)產(chǎn)業(yè)可持續(xù)發(fā)展的關(guān)鍵,電動(dòng)車(chē)輛由于其節(jié)能環(huán)保的優(yōu)勢(shì)成為汽車(chē)產(chǎn)業(yè)可持續(xù)發(fā)展的重要組成部分。對(duì)于電動(dòng)車(chē)輛而言,電池技術(shù)又是關(guān)乎其發(fā)展的一項(xiàng)重要因素。
2、電池的安全性與可靠性至關(guān)重要,在電池生產(chǎn)過(guò)程中需要對(duì)可能出現(xiàn)的缺陷進(jìn)行更加嚴(yán)格、準(zhǔn)確的管控。
技術(shù)實(shí)現(xiàn)思路
1、本技術(shù)旨在至少解決背景技術(shù)中存在的技術(shù)問(wèn)題之一。為此,本技術(shù)的一個(gè)目的在于提供一種圖像檢測(cè)模型及其訓(xùn)練方法、電池檢測(cè)方法、訓(xùn)練裝置、電子設(shè)備、電池生產(chǎn)設(shè)備、存儲(chǔ)介質(zhì)和程序產(chǎn)品,以提升對(duì)電池缺陷檢測(cè)的準(zhǔn)確性和可靠性。
2、本技術(shù)第一方面的實(shí)施例提供一種圖像檢測(cè)模型,包括骨干網(wǎng)絡(luò)、上采樣網(wǎng)絡(luò)和第一檢測(cè)層,上采樣網(wǎng)絡(luò)的輸入包括骨干網(wǎng)絡(luò)的輸出,骨干網(wǎng)絡(luò)被配置用于對(duì)所輸入的檢測(cè)圖像進(jìn)行下采樣,上采樣網(wǎng)絡(luò)被配置用于對(duì)骨干網(wǎng)絡(luò)所輸出的特征圖進(jìn)行上采樣,骨干網(wǎng)絡(luò)包括靠近圖像檢測(cè)模型的輸入端的網(wǎng)絡(luò)淺層,第一檢測(cè)層的輸入包括網(wǎng)絡(luò)淺層輸出的第一特征圖和上采樣網(wǎng)絡(luò)輸出的第二特征圖,第一檢測(cè)層被配置用于基于所輸入的第一特征圖和第二特征圖獲取檢測(cè)圖像的第一圖像檢測(cè)結(jié)果。
3、本技術(shù)實(shí)施例的技術(shù)方案中,由于在圖像檢測(cè)模型的第一檢測(cè)層的輸入包括網(wǎng)絡(luò)淺層輸出的第一特征圖和上采樣網(wǎng)絡(luò)輸出的第二特征圖,使得圖像檢測(cè)模型兼顧網(wǎng)絡(luò)淺層的高分辨率特征和上采樣網(wǎng)絡(luò)的深層語(yǔ)義特征,有效綜合利用不同尺度的信息,從而提升針對(duì)小目標(biāo)的檢測(cè)效果。
4、在一些實(shí)施例中,第一圖像檢測(cè)結(jié)果包括包圍檢測(cè)圖像中第一預(yù)設(shè)目標(biāo)的第一檢測(cè)框,第一檢測(cè)層被配置用于基于預(yù)設(shè)檢測(cè)框尺寸、第一特征圖和第二特征圖獲取檢測(cè)圖像的第一圖像檢測(cè)結(jié)果,其中,第一檢測(cè)框的尺寸與預(yù)設(shè)檢測(cè)框尺寸一致。由此,通過(guò)將模型輸出的檢測(cè)框約束至固定尺寸,從而能夠避免預(yù)測(cè)的檢測(cè)框因后處理的非極大抑制操作而被過(guò)濾掉,導(dǎo)致漏檢的問(wèn)題,實(shí)現(xiàn)對(duì)特定尺寸目標(biāo)的檢測(cè)。尤其是當(dāng)同類小目標(biāo)密集出現(xiàn)時(shí),通過(guò)約束預(yù)測(cè)的檢測(cè)框的尺寸能夠避免小目標(biāo)漏檢的問(wèn)題。
5、在一些實(shí)施例中,第一特征圖和第二特征圖兩者的尺度相同,能夠基于相同尺度的第一特征圖和第二特征圖直接進(jìn)行融合,提高圖像檢測(cè)模型的檢測(cè)效率。
6、在一些實(shí)施例中,骨干網(wǎng)絡(luò)包括多個(gè)依次連接的第一網(wǎng)絡(luò)層,多個(gè)第一網(wǎng)絡(luò)層包括網(wǎng)絡(luò)淺層和位于中間連接位置的中間網(wǎng)絡(luò)層,網(wǎng)絡(luò)淺層位于中間網(wǎng)絡(luò)層的靠近圖像檢測(cè)模型的輸入端的一側(cè)。通過(guò)設(shè)置網(wǎng)絡(luò)淺層為靠近輸入端的網(wǎng)絡(luò)層,從而能夠獲取圖像的高分辨率特征,有利于針對(duì)小目標(biāo)的檢測(cè)。
7、在一些實(shí)施例中,還包括第二檢測(cè)層,上采樣網(wǎng)絡(luò)包括多個(gè)第二網(wǎng)絡(luò)層,骨干網(wǎng)絡(luò)包括遠(yuǎn)離圖像檢測(cè)模型的輸入端的網(wǎng)絡(luò)深層,第二檢測(cè)層的輸入包括網(wǎng)絡(luò)深層輸出的第三特征圖和其中一個(gè)第二網(wǎng)絡(luò)層輸出的第四特征圖,第二檢測(cè)層被配置用于基于第三特征圖和第四特征圖獲取檢測(cè)圖像的第二圖像檢測(cè)結(jié)果,其中,第二圖像檢測(cè)結(jié)果包括包圍檢測(cè)圖像中第二預(yù)設(shè)目標(biāo)的第二檢測(cè)框,第二預(yù)設(shè)目標(biāo)的尺寸大于檢測(cè)圖像中第一預(yù)設(shè)目標(biāo)的尺寸。由于第二檢測(cè)層的輸入包括網(wǎng)絡(luò)深層輸出的第三特征圖和其中一個(gè)第二網(wǎng)絡(luò)層輸出的第四特征圖,通過(guò)第二檢測(cè)層獲取檢測(cè)圖像的第二圖像檢測(cè)結(jié)果,使得圖像檢測(cè)模型能夠?qū)Υ竽繕?biāo)具有良好的檢測(cè)效果,從而使得圖像檢測(cè)模型對(duì)大目標(biāo)和小目標(biāo)均具有良好的檢測(cè)效果,提高目標(biāo)檢測(cè)的完整性與全面性。
8、在一些實(shí)施例中,骨干網(wǎng)絡(luò)包括多個(gè)依次連接的第一網(wǎng)絡(luò)層,多個(gè)第一網(wǎng)絡(luò)層包括網(wǎng)絡(luò)深層和位于中間連接位置的中間網(wǎng)絡(luò)層,網(wǎng)絡(luò)深層位于中間網(wǎng)絡(luò)層的遠(yuǎn)離圖像檢測(cè)模型的輸入端的一側(cè)。通過(guò)設(shè)置網(wǎng)絡(luò)深層為遠(yuǎn)離輸入側(cè)的網(wǎng)絡(luò)層,從而能夠獲取圖像的深層語(yǔ)義特征,有利于針對(duì)大目標(biāo)的檢測(cè)。
9、本技術(shù)第二方面的實(shí)施例提供一種上述任一項(xiàng)的圖像檢測(cè)模型的訓(xùn)練方法,包括:獲取樣本電池檢測(cè)圖像;標(biāo)記樣本電池檢測(cè)圖像中包圍極片的端部的真實(shí)邊界框;將樣本電池檢測(cè)圖像輸入圖像檢測(cè)模型,并獲取圖像檢測(cè)模型所輸出的包圍極片的端部的預(yù)測(cè)邊界框;基于真實(shí)邊界框和預(yù)測(cè)邊界框計(jì)算損失值;以及基于損失值調(diào)整圖像檢測(cè)模型的參數(shù)。
10、本技術(shù)實(shí)施例的技術(shù)方案中,由于基于真實(shí)邊界框和預(yù)測(cè)邊界框之間的損失值,調(diào)整圖像檢測(cè)模型的參數(shù),從而使得預(yù)測(cè)邊界框接近真實(shí)邊界框,使得圖像檢測(cè)模型能夠充分學(xué)習(xí)極片端部的特征,實(shí)現(xiàn)對(duì)極片的端部的檢測(cè)。
11、在一些實(shí)施例中,獲取樣本電池的檢測(cè)圖像包括:獲取樣本電池在射線照射下的三維圖像;以及對(duì)三維圖像進(jìn)行切片,以得到二維的樣本電池檢測(cè)圖像。由此,圖像檢測(cè)模型能夠?qū)W習(xí)任意剖面的極片端部的特征,提升圖像檢測(cè)模型對(duì)極片端部檢測(cè)的泛化性。
12、在一些實(shí)施例中,對(duì)三維圖像進(jìn)行切片,以得到二維的樣本電池檢測(cè)圖像包括:對(duì)三維圖像進(jìn)行切片,以得到二維圖像;以及對(duì)二維圖像進(jìn)行預(yù)處理,以得到樣本電池檢測(cè)圖像,預(yù)處理包括去噪處理和對(duì)比度增強(qiáng)處理。通過(guò)對(duì)二維圖像進(jìn)行預(yù)處理,提高樣本電池檢測(cè)圖像的清晰度與精準(zhǔn)度,進(jìn)而提高圖像檢測(cè)模型對(duì)電池的訓(xùn)練效果。
13、本技術(shù)第三方面的實(shí)施例提供一種基于上述任一項(xiàng)圖像檢測(cè)模型的電池檢測(cè)方法,包括:獲取電池檢測(cè)圖像;將電池檢測(cè)圖像輸入圖像檢測(cè)模型,并獲取圖像檢測(cè)模型所輸出的電池的第一圖像檢測(cè)結(jié)果;以及基于第一圖像檢測(cè)結(jié)果對(duì)電池進(jìn)行檢測(cè)。
14、本技術(shù)實(shí)施例的技術(shù)方案中,由于在圖像檢測(cè)模型的第一檢測(cè)層的輸入包括網(wǎng)絡(luò)淺層輸出的第一特征圖和上采樣網(wǎng)絡(luò)輸出的第二特征圖,使得圖像檢測(cè)模型兼顧網(wǎng)絡(luò)淺層的高分辨率特征和上采樣網(wǎng)絡(luò)的深層語(yǔ)義特征,有效綜合利用不同尺度的信息,從而提升針對(duì)電池的尺寸較小特征的檢測(cè)效果,降低漏檢的風(fēng)險(xiǎn),從而提高對(duì)電池的檢測(cè)效果。
15、在一些實(shí)施例中,圖像檢測(cè)模型的第一檢測(cè)層被配置用于基于預(yù)設(shè)檢測(cè)框尺寸、第一特征圖和第二特征圖獲取電池檢測(cè)圖像的第一圖像檢測(cè)結(jié)果,第一圖像檢測(cè)結(jié)果包括包圍電池檢測(cè)圖像中極片的端部的第一檢測(cè)框,其中,第一檢測(cè)框的尺寸與預(yù)設(shè)檢測(cè)框尺寸一致,預(yù)設(shè)檢測(cè)框尺寸為基于電池的極片的端部的尺寸。由于預(yù)設(shè)檢測(cè)框尺寸為基于電池的極片的端部的尺寸,第一檢測(cè)框的尺寸與預(yù)設(shè)檢測(cè)框尺寸一致,有利于提升圖像檢測(cè)模型對(duì)電池極片的端部的識(shí)別準(zhǔn)確度,降低漏檢風(fēng)險(xiǎn),提高圖像檢測(cè)模型的檢測(cè)效果。
16、在一些實(shí)施例中,極片包括陰極片和陽(yáng)極片,基于第一圖像檢測(cè)結(jié)果對(duì)電池進(jìn)行檢測(cè)包括:基于分別包圍電池的同一極片的兩個(gè)端部的兩個(gè)第一檢測(cè)框,確定包括極片的目標(biāo)檢測(cè)框;以及基于包圍陰極片的目標(biāo)檢測(cè)框和陽(yáng)極片的目標(biāo)檢測(cè)框,確定陰極片和陽(yáng)極片的目標(biāo)錯(cuò)位量,目標(biāo)錯(cuò)位量為基于沿陰極片的寬度方向和/或長(zhǎng)度方向上陰極片和陽(yáng)極片的錯(cuò)位量。通過(guò)包圍極片端部的第一檢測(cè)框確定包圍極片的目標(biāo)檢測(cè)框,從而確定陰極片和陽(yáng)極片的目標(biāo)錯(cuò)位量,實(shí)現(xiàn)對(duì)電池極片錯(cuò)位量的檢測(cè),通過(guò)提升對(duì)極片端部的識(shí)別準(zhǔn)確度,通過(guò)極片端部確定包圍極片的目標(biāo)檢測(cè)框,能夠提高對(duì)電池極片錯(cuò)位量的檢測(cè)效果。
17、在一些實(shí)施例中,電池檢測(cè)方法包括:對(duì)獲取的電池檢測(cè)圖像進(jìn)行預(yù)處理,預(yù)處理包括去噪處理、對(duì)比度增強(qiáng)處理中的至少一種,并且將經(jīng)過(guò)預(yù)處理的電池檢測(cè)圖像輸入圖像檢測(cè)模型。通過(guò)對(duì)獲取的電池檢測(cè)圖像進(jìn)行預(yù)處理,能夠減少電池檢測(cè)圖像中背景噪聲對(duì)目標(biāo)識(shí)別造成的干擾,從而改善電池檢測(cè)圖像的檢測(cè)效果。
18、在一些實(shí)施例中,對(duì)比度增強(qiáng)至少包括隨機(jī)翻轉(zhuǎn)、尺度變換、圖像平移、添加噪聲中的至少一種。通過(guò)對(duì)獲取的電池檢測(cè)圖像進(jìn)行對(duì)比度增強(qiáng),提高電池檢測(cè)圖像的清晰程度,有利于更好地對(duì)電池檢測(cè)圖像進(jìn)行檢測(cè),提高檢測(cè)效果。
19、在一些實(shí)施例中,電池檢測(cè)方法包括:響應(yīng)于電池的檢測(cè)結(jié)果為不合格,將電池剔除或復(fù)檢。通過(guò)將電池的檢測(cè)結(jié)果(例如極片錯(cuò)位量)為不合格的電池剔除或復(fù)檢,提高電池的安全性和可靠性。
20、本技術(shù)第四方面的實(shí)施例提供一種圖像檢測(cè)模型的訓(xùn)練裝置,訓(xùn)練裝置包括:獲取單元,用于獲取樣本電池檢測(cè)圖像;標(biāo)記單元,用于標(biāo)記樣本電池檢測(cè)圖像中包圍極片的端部的真實(shí)邊界框;計(jì)算單元,用于基于圖像檢測(cè)模型獲取圖像檢測(cè)模型所輸出的包圍極片的端部的預(yù)測(cè)邊界框,并基于真實(shí)邊界框和預(yù)測(cè)邊界框計(jì)算損失值;以及調(diào)整單元,用于基于損失值調(diào)整圖像檢測(cè)模型的參數(shù)。
21、本技術(shù)實(shí)施例的技術(shù)方案中,通過(guò)訓(xùn)練裝置的計(jì)算單元基于真實(shí)邊界框和預(yù)測(cè)邊界框之間的損失值,調(diào)整圖像檢測(cè)模型的參數(shù),從而使得預(yù)測(cè)邊界框接近真實(shí)邊界框,圖像檢測(cè)模型能夠充分學(xué)習(xí)極片端部的特征,實(shí)現(xiàn)對(duì)極片的端部的檢測(cè)。
22、本技術(shù)第五方面的實(shí)施例提供一種電子設(shè)備,電子設(shè)備包括:至少一個(gè)處理器;以及與至少一個(gè)處理器通信連接的存儲(chǔ)器;其中,存儲(chǔ)器存儲(chǔ)有能夠被至少一個(gè)處理器執(zhí)行的指令,指令被至少一個(gè)處理器執(zhí)行,以使至少一個(gè)處理器能夠執(zhí)行上述任一項(xiàng)的方法。
23、本技術(shù)第六方面的實(shí)施例提供一種電池生產(chǎn)設(shè)備,包括上述的電子設(shè)備。
24、本技術(shù)第七方面的實(shí)施例提供一種計(jì)算機(jī)可讀存儲(chǔ)介質(zhì),存儲(chǔ)有計(jì)算機(jī)程序,計(jì)算機(jī)程序被處理器執(zhí)行時(shí)實(shí)現(xiàn)上述任一項(xiàng)的方法。
25、本技術(shù)第八方面的實(shí)施例提供一種計(jì)算機(jī)程序產(chǎn)品,包括計(jì)算機(jī)程序,該計(jì)算機(jī)程序被處理器執(zhí)行時(shí)實(shí)現(xiàn)上述任一項(xiàng)的方法。
26、上述說(shuō)明僅是本技術(shù)技術(shù)方案的概述,為了能夠更清楚了解本技術(shù)的技術(shù)手段,而可依照說(shuō)明書(shū)的內(nèi)容予以實(shí)施,并且為了讓本技術(shù)的上述和其它目的、特征和優(yōu)點(diǎn)能夠更明顯易懂,以下特舉本技術(shù)的具體實(shí)施方式。