[發明專利]基于YOLOv4-tiny的插板閥開度檢測方法有效
| 申請號: | 202011502843.3 | 申請日: | 2020-12-17 |
| 公開(公告)號: | CN112561885B | 公開(公告)日: | 2023-04-18 |
| 發明(設計)人: | 李明;鹿朋;朱美強;梁健;王軍 | 申請(專利權)人: | 中國礦業大學 |
| 主分類號: | G06T7/00 | 分類號: | G06T7/00;G06T7/73;G06V10/774;G06V10/762;G06V10/764;G06V10/82;G06N3/04 |
| 代理公司: | 無錫知更鳥知識產權代理事務所(普通合伙) 32468 | 代理人: | 張濤 |
| 地址: | 221116 江蘇省徐州*** | 國省代碼: | 江蘇;32 |
| 權利要求書: | 查看更多 | 說明書: | 查看更多 |
| 摘要: | |||
| 搜索關鍵詞: | 基于 yolov4 tiny 插板 閥開度 檢測 方法 | ||
1.一種基于YOLOv4-tiny的插板閥開度檢測方法,其特征是,所述插板閥開度檢測方法包括如下步驟:
步驟S1、獲取插板閥的初始樣本數據集,并對所獲取的初始樣本數據集進行所需的預處理,以得到插板閥的標準樣本數據集;
步驟S2、對上述插板閥的標準樣本數據集進行物體類別標注與位置信息的標注,以得到插板閥的標準樣本標注數據集,且對插板閥的標準樣本標注數據集進行劃分,以得到訓練數據集、驗證數據集與測試數據集;
步驟S3、讀取YOLOv4-tiny的配置文件,并配置所述YOLOv4-tiny配置文件中的網絡層參數信息,以得到YOLOv4-tiny基本檢測模型;對于所述YOLOv4-tiny基本檢測模型,使用CSPDarknet53作為backbone進行主要特征提取器,使用子階段sub-stage作為neck進行特征融合,使用YOLO作為head進行目標位置與類別信息的預測,且將構建的空間金字塔池化模塊分別與索引為m的convolution模塊、索引為m+1的convolution模塊連接;
步驟S4、利用訓練數據集對上述的YOLOv4-tiny基本檢測模型進行訓練,當訓練達到設定的次數或在驗證數據集上出現過擬合時停止訓練,在每次的訓練中,更新YOLOv4-tiny基本檢測模型的權重信息,以在每次訓練后均能得到一相應的YOLOv4-tiny基本檢測訓練模型;
步驟S5、利用測試數據集對上述步驟S4中得到所有的YOLOv4-tiny基本檢測訓練模型進行測試;測試后,任一YOLOv4-tiny基本檢測訓練模型得到一相應的多尺度預測信息,利用每個YOLOv4-tiny基本檢測訓練模型的多尺度預測信息確定對應YOLOv4-tiny基本檢測訓練模型的檢測精度,將檢測精度最高的YOLOv4-tiny基本檢測訓練模型作為YOLOv4-tiny目標檢測模型;
步驟S6、利用YOLOv4-tiny目標檢測模型對待計算開度的插板閥圖像進行處理,以得到所述插板閥圖像的開度;
步驟S2中,對標準樣本數據集進行物體類別標準包括對插板外框的標注以及對插板的標注;標注時,采用LabelImage標注工具進行標注;在標注后,生成VOC2007標準數據格式的數據集;劃分得到的訓練數據集、驗證數據集、測試數據集相應的圖像數量比例為8:1:1;
步驟S6中,插板閥圖像的開度O為:
其中,xfull為插板全開的位置,xempty為插板全閉的位置,x為插板的當前位置。
2.根據權利要求1所述的基于YOLOv4-tiny的插板閥開度檢測方法,其特征是:步驟S1中,插板閥的初始樣本數據集包括插板閥在不同場景、不同時段、不同角度和/或不同光照下的圖像序列;對初始樣本數據集的預處理包括數據增廣處理和/或數據增強處理。
3.根據權利要求1所述的基于YOLOv4-tiny的插板閥開度檢測方法,其特征是,步驟S4包括如下步驟:
步驟S4.1、對訓練數據集,采用K-means聚類方法對插板閥內的插板、插板外框相對應的尺寸進行聚類,以得到M種的先驗框;
步驟S4.2、配置YOLOv4-tiny基本檢測模型的超參數,并對YOLOv4-tiny基本檢測模型在訓練過程中,隨機改變訓練數據集中輸入到YOLOv4-tiny基本檢測模型中相應圖像的尺寸,以實現對YOLOv4-tiny基本檢測模型的多尺度的訓練;
步驟S4.3、更新YOLOv4-tiny基本檢測模型的權重信息,使得更新權重信息后的YOLOv4-tiny基本檢測模型利用訓練數據集進行訓練時,損失函數的數值總體上呈下降趨勢;
步驟4.4、當訓練達到設定的次數或在驗證數據集上出現過擬合時停止訓練,根據更新后不同的權重信息,能在每次訓練后均能得到一相應的YOLOv4-tiny基本檢測訓練模型。
該專利技術資料僅供研究查看技術是否侵權等信息,商用須獲得專利權人授權。該專利全部權利屬于中國礦業大學,未經中國礦業大學許可,擅自商用是侵權行為。如果您想購買此專利、獲得商業授權和技術合作,請聯系【客服】
本文鏈接:http://www.17sss.com.cn/pat/books/202011502843.3/1.html,轉載請聲明來源鉆瓜專利網。





