关于 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_depth构建SPD-Conv主干网络 ;
  5. 修改 ultralytics/yolo/cfg/default.yaml 文件的 '--model' 默认参数,或者直接使用指令&#

猜你喜欢

转载自blog.csdn.net/weixin_43694096/article/details/131020927
py