一種基于EfficientNet的螺母表面缺陷分類方法
基本信息
| 申請(qǐng)?zhí)?/td> | CN202110289071.8 | 申請(qǐng)日 | - |
| 公開（公告）號(hào) | CN113052809B | 公開（公告）日 | 2021-12-10 |
| 申請(qǐng)公布號(hào) | CN113052809B | 申請(qǐng)公布日 | 2021-12-10 |
| 分類號(hào) | G06T7/00;G06T3/60;G06T7/11;G06T7/136;G06T7/194;G06T7/187;G06K9/62 | 分類 | 計(jì)算；推算；計(jì)數(shù); |
| 發(fā)明人 | 李子杰;程坦;劉濤;呂劍 | 申請(qǐng)（專利權(quán)）人 | 中科海拓（無(wú)錫）科技有限公司 |
| 代理機(jī)構(gòu) | 安徽善安知識(shí)產(chǎn)權(quán)代理事務(wù)所（特殊普通合伙） | 代理人 | 陳庭 |
| 地址 | 214000 江蘇省無(wú)錫市濱湖區(qū)新城置業(yè)大廈20樓 | ||
| 法律狀態(tài) | - | ||
摘要
| 摘要 | 本發(fā)明公開了一種基于EfficientNet的螺母表面缺陷分類方法，包括步驟一：采集訓(xùn)練樣本，制作數(shù)據(jù)集；通過(guò)數(shù)據(jù)集訓(xùn)練EfficientNet模型，模型訓(xùn)練收斂后，再進(jìn)行螺母表面分類；通過(guò)使用EfficientNet對(duì)模型規(guī)模進(jìn)行自調(diào)整，避免人工調(diào)整費(fèi)時(shí)，且難以得到最優(yōu)解；步驟二：采集目標(biāo)圖像，對(duì)目標(biāo)圖像進(jìn)行裁剪；步驟三：修改損失函數(shù)為focal loss，調(diào)整正負(fù)樣本權(quán)重，提升模型精度；免了正負(fù)樣本不均衡對(duì)模型的影響；步驟四：圖像預(yù)處理，制作模板圖像對(duì)輸入圖像進(jìn)行旋轉(zhuǎn)矯正，并做差得到差值圖像，將三者融合為三通道圖像作為模型輸入圖像；增強(qiáng)模型對(duì)生產(chǎn)環(huán)境的適應(yīng)能力，提升模型的泛化能力，步驟五：模型推理，使用EfficientNet模型對(duì)輸入圖像進(jìn)行推理；步驟六：推理圖像輸出到上位機(jī)顯示。 |
