亚洲一区欧美在线,日韩欧美视频免费观看,色戒的三场床戏分别是在几段,欧美日韩国产在线人成

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

作者簡(jiǎn)介:

通訊作者:

中圖分類號(hào):

基金項(xiàng)目:

中國(guó)水利水電科學(xué)研究院“五大人才”計(jì)劃專項(xiàng)(SD0145B032021)和國(guó)家自然科學(xué)基金項(xiàng)目(51809291)


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

Fund Project:

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

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

    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.

    參考文獻(xiàn)
    相似文獻(xiàn)
    引證文獻(xiàn)
引用本文

李健源,柳春娜,盧曉春,吳必朗.基于改進(jìn)YOLOv5s和TensorRT部署的魚道過魚監(jiān)測(cè)[J].農(nóng)業(yè)機(jī)械學(xué)報(bào),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.

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