最近 YOLOv8
的官方项目又迎来了一个大更新,这次更新对基础不好的同学影响可能比较大,
这次更新主要就是将原本的 ultralytics/nn/modules.py
拆分成了以下6
个文件:
_init_.py
block.py
conv.py
head.py
trasnformer.py
utils.py
有的同学可能不知道该加哪里了,我这里给大家说一下
我们原本的添加模块的修改思路如下,以 space_to_depth
举例子:
- 在
ultralytics/models/v8
文件夹下新建一个yolov8-SPD-Conv.yaml
; - 将本文上面提供的
SPD-Conv
代码添加到ultralytics/nn/modules.py
文件末尾; - 将
SPD-Conv
这个类的名字加入到ultralytics/nn/tasks.py
中; - 修改
yolov8-SPD-Conv.y