文章大纲
YOLOv8 源码解析,如何进行分类任务?
https://github.com/ultralytics/ultralytics/issues?q=classification
准备工作:visual code 调试源码
vscode的调试配置里有个重要的选项没配置,就是"justMyCode": false。这个选项默认是true,是进不了第三方包源码的,所以要改成false。
本来想找一个调用堆栈可视化的软件,但是发现这个好像不是很好用:
https://marketplace.visualstudio.com/items?itemName=hediet.debug-visualizer
pip install vscodedebugvisualizer
那我们直接从模型入手,看看结构:
yolov8 图像分类的网络结构很简单,就是Conv 和C2f 的 类似卷积的重复。不同模型就是参数量级的不同。