本發明提供一種基于改進yolov5的連鑄坯裂紋缺陷識別方法,涉及冶金和圖像識別技術領域。該方法首先構建連鑄坯裂紋缺陷數據集,并進行標注,得到連鑄坯裂紋缺陷數據訓練集;再將CBAM模塊和yolov5網絡模型結合,構建CBAMC3?yolov5網絡模型;然后利用連鑄坯裂紋缺陷數據訓練集對CBAMC3?yolov5網絡模型進行訓練;最后使用訓練好的CBAMC3?yolov5網絡模型對待識別的連鑄坯裂紋缺陷圖像進行裂紋缺陷識別。該方法通過改進yolov5算法并結合連鑄坯裂紋缺陷圖像數據集,成功識別連鑄坯裂紋缺陷圖像,得到裂紋缺陷的位置信息與分類信息。解決了人工識別主觀性強、工作強度大等明顯缺點。
聲明:
“基于改進yolov5的連鑄坯裂紋缺陷識別方法” 該技術專利(論文)所有權利歸屬于技術(論文)所有人。僅供學習研究,如用于商業用途,請聯系該技術所有人。
我是此專利(論文)的發明人(作者)