ass日本风韵熟妇pics男人扒开女人屁屁桶到爽|扒开胸露出奶头亲吻视频|邻居少妇的诱惑|人人妻在线播放|日日摸夜夜摸狠狠摸婷婷|制服 丝袜 人妻|激情熟妇中文字幕|看黄色欧美特一级|日本av人妻系列|高潮对白av,丰满岳妇乱熟妇之荡,日本丰满熟妇乱又伦,日韩欧美一区二区三区在线

基于改進YOLOv5s和TensorRT部署的魚道過魚監(jiān)測
CSTR:
作者:
作者單位:

作者簡介:

通訊作者:

中圖分類號:

基金項目:

中國水利水電科學研究院“五大人才”計劃專項(SD0145B032021)和國家自然科學基金項目(51809291)


Fish Passage Monitoring Based on Improved YOLOv5s and TensorRT Deployment
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 圖/表
  • |
  • 訪問統(tǒng)計
  • |
  • 參考文獻
  • |
  • 相似文獻
  • |
  • 引證文獻
  • |
  • 資源附件
  • |
  • 文章評論
    摘要:

    為實現(xiàn)在復雜水體下對魚道過魚進行監(jiān)測,,提出了一種基于YOLOv5s的改進模型,,并用TensorRT部署應(yīng)用于某水電站魚道現(xiàn)場,。首先,,針對水下圖像模糊、目標檢測困難的問題,,提出了將Swin Transformer(STR)作為檢測層,,提升了模型對目標的檢測能力;其次,,針對魚群密集,、圖像信息少的問題,將Efficient channel attention(ECA)注意力機制作為主干特征提取網(wǎng)絡(luò)C3結(jié)構(gòu)的Bottleneck,,減少了計算參數(shù)并提升了檢測精度,;然后,針對檢測目標定位差,、回歸不收斂的問題,,將Focal and efficient IOU loss(FIOU)作為模型損失函數(shù),優(yōu)化了模型整體性能,;最后將模型部署在TensorRT框架進行優(yōu)化,,處理速度得到了大幅度提升?;趯嶋H采集的復雜水體數(shù)據(jù)集進行實驗,,結(jié)果表明,本文算法mAP為91.9%,,單幅圖像處理時間為10.4ms,,在相同條件下,,精度比YOLOv5s提升4.8個百分點,處理時間減少0.4ms,。模型使用TensorRT部署后單幅圖像推理時間達到2.3ms,,在不影響檢測精度的前提下,推理速度提高4.5倍,。綜上,本文算法模型在保證快速檢測的基礎(chǔ)上,,具有較高的準確性,,更適用于復雜水體下魚道過魚監(jiān)測。

    Abstract:

    In order to realize the detection of fish passage in fishway under complex water, an improved model based on YOLOv5s was proposed, and it was deployed and applied to the fishway site of one hydropower station with TensorRT. Firstly, in view of the problems of underwater image blur and target detection difficulty, the Swin Transformer (STR) was proposed as the detection layer, which improved the detection ability of the model for targets. Secondly, in view of the problem of dense fish and little image information, the efficient channel attention (ECA) attention mechanism was used as the Bottleneck of the backbone feature extraction network C3 structure, which reduced the calculation parameters and improved the detection accuracy. Then aiming at the problem of detection target positioning error and non convergence of regression, taking focal and efficient IOU loss (FIOU) as the loss function of the model to optimize the overall performance of the model. Finally, the model was deployed in TensorRT framework for optimization, and the processing speed was greatly improved. Based on the actual collection of complex water body data set, the experiment results showed that the algorithm mAP was 91.9%, and the processing time of a single image was 10.4ms. Under the same conditions, the precision was 4.8 percentage points higher than that of YOLOv5s, and the processing time was 0.4ms. After the model was deployed with TensorRT, the reasoning speed reached 2.3ms/img, a 4.5 times improvement in reasoning speed without affecting detection accuracy. In conclusion, the algorithm model had good effectiveness and superiority, and it was more suitable for fish passage detection in complex water bodies.

    參考文獻
    相似文獻
    引證文獻
引用本文

李健源,柳春娜,盧曉春,吳必朗.基于改進YOLOv5s和TensorRT部署的魚道過魚監(jiān)測[J].農(nóng)業(yè)機械學報,2022,53(12):314-322. LI Jianyuan, LIU Chunna, LU Xiaochun, WU Bilang. Fish Passage Monitoring Based on Improved YOLOv5s and TensorRT Deployment[J]. Transactions of the Chinese Society for Agricultural Machinery,2022,53(12):314-322.

復制
分享
文章指標
  • 點擊次數(shù):
  • 下載次數(shù):
  • HTML閱讀次數(shù):
  • 引用次數(shù):
歷史
  • 收稿日期:2022-09-15
  • 最后修改日期:
  • 錄用日期:
  • 在線發(fā)布日期: 2022-10-18
  • 出版日期:
文章二維碼