YOLOv5 YOLOv7 YOLOv8 のモデルテストの比較と結論

YOLOv5 YOLOv7 YOLOv8 のモデルテストの比較と結論

背景情報

独自にトレーニングされたモデルを使用することに基づいて、yolov8 が推論テストを行うと、後処理時間が推論時間よりも長くなるため、この調査が行われました。val および予測速度の列で、Yolov8 の後処理時間が 16 ミリ秒以上に達していることを初めて発見しました。【頭を高く上げてください。とても恥ずかしいです。最終的に、予測と検証中に、nvidia-smi によって表示されるグラフィックス カード情報が別の Python プログラムによって占有され、20G+ を占有していることがわかりました。背後に他の Python プログラムが存在しない場合、Speed表 2. 各モデルのトレーニング結果の比較 に示すように結果が表示されましたが、おそらく nms を実行すると CPU の占有率が高くなり、ポストプロセスの速度に影響を与えると思われます。nms の実際の速度は、推論の速度よりもはるかに遅くなります。

トレーニングと検証テストには 9285:1985:1987 を使用しました。

(1) トレーニングプラットフォーム情報

表1 トレーニングプラットフォームの関連情報

システム: Ubuntu 20.04
ドライバーのバージョン: 525.78.01

おすすめ

転載: blog.csdn.net/qq_42835363/article/details/131852294