基于MaskRCNN的圖像分割模型訓(xùn)練方法及粒徑檢測方法

基本信息

申請?zhí)?/td> CN202110771044.4 申請日 -
公開（公告）號(hào) CN113408478A 公開（公告）日 2021-09-17
申請公布號(hào) CN113408478A 申請公布日 2021-09-17
分類號(hào) G06K9/00（2006.01）I;G06K9/34（2006.01）I;G06K9/46（2006.01）I;G06K9/62（2006.01）I;G06T5/00（2006.01）I;G06T7/12（2017.01）I;G06T7/13（2017.01）I;G06T7/80（2017.01）I;G06N3/04（2006.01）I;G06N3/08（2006.01）I 分類 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 甘化新;劉洪;張振華;胡吉鋒;陳勝平;陳林空 申請（專利權(quán)）人 深圳市宏申工業(yè)智能有限公司
代理機(jī)構(gòu) 北京維正專利代理有限公司 代理人 吳珊
地址 518000廣東省深圳市寶安區(qū)石巖街道羅租社區(qū)黃峰嶺工業(yè)區(qū)源進(jìn)廠一層至四層，廠房1一層至二層，廠房5一層至二層
法律狀態(tài) -

摘要

摘要 本申請公開了基于Mask RCNN的圖像分割模型訓(xùn)練方法及粒徑檢測方法，訓(xùn)練方法包括：獲取石子的檢測圖像和樣例圖像；將檢測圖像輸入CNN中，CNN對檢測圖像進(jìn)行特征提取，以得到特征圖；通過RPN生成錨框，并根據(jù)RPN回歸分支的輸出結(jié)果對錨框進(jìn)行修正，且選擇預(yù)定數(shù)量的錨框以形成建議窗口；把建議窗口映射到CNN最后一層的特征圖上，通過RoI Align層使每個(gè)建議窗口生成固定尺寸的RoI特征圖；利用全連接層對RoI分類與邊框回歸，通過FCN生成每個(gè)RoI的mask，分割得到石子輪廓；計(jì)算石子輪廓與樣例圖像標(biāo)記的真實(shí)輪廓之間的偏移量，根據(jù)偏移量對Mask RCNN神經(jīng)網(wǎng)絡(luò)的參數(shù)進(jìn)行更新。本申請具有提高粒度檢測精準(zhǔn)度的效果。