YOLOv8+OpenCV實現(xiàn)DM碼定位檢測與解析

YOLOv8是YOLO系列模型的最新王者，各種指標全面超越現(xiàn)有對象檢測與實例分割模型，借鑒了YOLOv5、YOLOv6、YOLOX等模型的設(shè)計優(yōu)點，全面提升改進YOLOv5的模型結(jié)構(gòu)基礎(chǔ)上實現(xiàn)同時保持了YOLOv5工程化簡潔易用的優(yōu)勢。

數(shù)據(jù)準備

工業(yè)碼（DataMatrix碼簡稱DM碼）檢測與識別是機器視覺最常用得場景之一，從DM碼的定位、質(zhì)量評估、預處理、檢測與ROI截取、解析識別。每個環(huán)節(jié)都有很多技術(shù)細節(jié)，以DM碼的發(fā)現(xiàn)于定位來說，早期都是基于傳統(tǒng)的圖像分析與尋邊算法實現(xiàn)DM碼定位，現(xiàn)在基本上都是基于輕量化的深度學習對象檢測模型來完成。本文的數(shù)據(jù)集是來自一個網(wǎng)友的貢獻，它的DM碼圖像。

傳統(tǒng)方法分析相對來說比較容易收到其它因素干擾，導致解碼率不高，因此我從他給我的2000多張圖像中挑選出200張，用OpenMV做好了標注，制作生成了數(shù)據(jù)集。

模型訓練與量化部署

然后基于YOLOv8對象檢測模型，使用遷移學習方式訓練了一個DM碼檢測模型，訓練結(jié)果如下：

把訓練生成的best.pt模型導出ONNX格式模型，再進一步量化為INT8，最終模型大小只有3MB，在端側(cè)板子上都可以部署運行，然后運行測試。

采用ZXING開源解碼結(jié)果

這其中最重要的就是定位查找DM碼，使用YOLOv8訓練一個自定義對象檢測模型，可以很輕松的搞定，然后基于ROI使用OpenCV完成預處理之后即可解碼。