基于神經(jīng)網(wǎng)絡(luò)的垃圾分類方法及裝置

基本信息

申請?zhí)?/td> CN202110594291.1 申請日 -
公開（公告）號 CN113496254A 公開（公告）日 2021-10-12
申請公布號 CN113496254A 申請公布日 2021-10-12
分類號 G06K9/62（2006.01）I;G06N3/04（2006.01）I;G06N3/08（2006.01）I 分類 計算；推算；計數(shù);
發(fā)明人 王鐸 申請（專利權(quán)）人 聞泰通訊股份有限公司
代理機構(gòu) 北京開陽星知識產(chǎn)權(quán)代理有限公司 代理人 唐博
地址 314006浙江省嘉興市南湖區(qū)亞中路777號（嘉興科技城）
法律狀態(tài) -

摘要

摘要 本申請涉及一種基于神經(jīng)網(wǎng)絡(luò)的垃圾分類方法、裝置、計算機設(shè)備和計算機可讀存儲介質(zhì)。所述方法包括：獲取訓練樣本集，所述訓練樣本集包括多個樣本垃圾圖像以及所述多個樣本垃圾圖像中各個樣本垃圾圖像對應的類別標簽；加載預訓練分類網(wǎng)絡(luò)模型，所述預訓練分類網(wǎng)絡(luò)模型為基于源數(shù)據(jù)對卷積神經(jīng)網(wǎng)絡(luò)模型進行訓練得到的圖像分類模型；基于遷移學習，通過所述訓練樣本集對所述預訓練分類網(wǎng)絡(luò)模型的輸出層網(wǎng)絡(luò)結(jié)構(gòu)進行訓練，獲取垃圾分類模型；所述輸出層網(wǎng)絡(luò)結(jié)構(gòu)包括至少一個全連接層；根據(jù)所述垃圾分類模型進行垃圾分類。采用本方法構(gòu)建小樣本的垃圾分類模型，減少參數(shù)數(shù)量，降低計算的復雜度，從而能夠?qū)崿F(xiàn)垃圾的準確、快速分類。