基于小樣本機(jī)器學(xué)習(xí)與超參數(shù)優(yōu)化的水下目標(biāo)檢測方法

基本信息

申請?zhí)?/td> CN202110442031.2 申請日 -
公開（公告）號 CN113239980B 公開（公告）日 2022-07-05
申請公布號 CN113239980B 申請公布日 2022-07-05
分類號 G06K9/62（2022.01）I;G06N3/02（2006.01）I 分類 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 李新宇;孫晨;文龍;萬宇森 申請（專利權(quán)）人 中國地質(zhì)大學(xué)（武漢）
代理機(jī)構(gòu) 武漢知產(chǎn)時代知識產(chǎn)權(quán)代理有限公司 代理人 -
地址 430000湖北省武漢市洪山區(qū)珞喻路1037號
法律狀態(tài) -

摘要

摘要 本發(fā)明提供了一種基于小樣本機(jī)器學(xué)習(xí)與超參數(shù)優(yōu)化的水下目標(biāo)檢測方法，包括：構(gòu)建基于Cascade?RCNN的小樣本目標(biāo)檢測模型；預(yù)訓(xùn)練數(shù)據(jù)集；利用預(yù)訓(xùn)練數(shù)據(jù)集對小樣本目標(biāo)檢測模型進(jìn)行預(yù)訓(xùn)練，獲得小樣本目標(biāo)檢測模型的預(yù)訓(xùn)練權(quán)重θ′；構(gòu)建待檢測的目標(biāo)數(shù)據(jù)集；將目標(biāo)數(shù)據(jù)集劃分為有標(biāo)記的支持集和無標(biāo)記的查詢集；對目標(biāo)數(shù)據(jù)集進(jìn)行預(yù)處理；對小樣本目標(biāo)檢測模型進(jìn)行微調(diào)，得到最終訓(xùn)練完成的小樣本目標(biāo)檢測模型；采用基于TPE的貝葉斯優(yōu)化模型對訓(xùn)練完成的小樣本目標(biāo)檢測模型進(jìn)行超參數(shù)優(yōu)化，得到優(yōu)化后的小樣本目標(biāo)檢測模型；輸入預(yù)處理后的查詢集至優(yōu)化后的目標(biāo)檢測模型，得到目標(biāo)檢測結(jié)果。本發(fā)明實(shí)現(xiàn)了不依賴于大樣本、可以自動調(diào)節(jié)超參數(shù)的目標(biāo)檢測。