摘要:針對堆疊密集的堆垛貨箱出現的漏檢情況以及難以分割出每個(gè)貨箱的精確邊緣而造成的難以準確抓取的問(wèn)題,對深度學(xué)習實(shí)例分割算法YOLACT進(jìn)行了相應的改進(jìn)。首先使用工業(yè)相機采集貨箱的堆垛圖像,然后利用Labelme標注圖像制作數據集,并且通過(guò)數據增強方法擴充數據集。接著(zhù)為了提高模型的分割準確率,分別對掩碼真值和YOLACT中的原型掩碼輸出分支(Protonet)的預測掩碼使用Canny邊緣檢測算子,并取二者的二值交叉熵損失作為損失函數加入到原網(wǎng)絡(luò )中訓練。最后再使用訓練好的最優(yōu)模型對測試集圖像數據進(jìn)行試驗,結果表明,改進(jìn)后的模型預測掩碼mAP0.5:0.95可以達到0.543,比原模型提高2.2%,同時(shí)貨箱邊緣的分割精度也得到了一定的提升,模型推理速度可達10.2幀/秒,可以滿(mǎn)足精度要求和生產(chǎn)節拍要求。