YOLOv8 について modules.py を _init_.py に分割 block.py conv.py head.py trasnformer.py utils.py

YOLOv8最近の公式プロジェクトにより大きなアップデートが行われましたが、このアップデートは基礎が不十分な学生にとってより大きな影響を与える可能性があります。

この更新は主に、オリジナルultralytics/nn/modules.pyを次の6ファイルに分割することです。

  1. _init_.py
  2. block.py
  3. conv.py
  4. head.py
  5. trasnformer.py
  6. utils.py

どこに追加すればよいかわからない学生もいるかもしれません。ここで説明します。


モジュールを追加するための私たちの独自の変更アイデアは、space_to_depth例。

  1. ultralytics/models/v8フォルダーの下に新しいものを作成しますyolov8-SPD-Conv.yaml
  2. この記事の上に記載されているSPD-Convコードをultralytics/nn/modules.pyファイルの最後に追加します。
  3. SPD-Convこのクラスの名前ultralytics/nn/tasks.pyを ; に追加します。
  4. を変更し、バックボーン ネットワークの構築yolov8-SPD-Conv.yamlに使用します。space_to_depthSPD-Conv
  5. ultralytics/yolo/cfg/default.yamlファイルのデフォルトのパラメータを変更する'--model'か、コマンド を直接使用します。

おすすめ

転載: blog.csdn.net/weixin_43694096/article/details/131020927