【DL】YOLOv8模型中的train.py

YOLOv8项目中有两个地方可以执行train.py

最外层的train.py

（现在想起来这里好像是我自己写的训练脚本，原项目里没有）


YOLO()构造函数默认构造的是目标检测的模型，也可以在方法的参数中指定task类型

models/yolo中的train.py

这里的每个文件夹中定义了相对应任务中的训练脚本

在这里是通过这种方式获取任务类型的
这个nn/tasks.py是在项目中写好的python文件
这个文件定义了yolo对于不同任务定义的不同模型结构（前向传播流程）

在这里调用这个变量

总结

在yolo中，不同任务的选择是在创建模型时选择对应的任务模板来确定的

补充1

容易发现，yaml中只有backbone和head部分的模块信息。
损失函数部分应该是定义在推理函数中。