【DL】YOLOv8模型中的train.py

合集 - 深度学习(6)

1.【pytorch】向量点积和向量乘积2024-06-202.【pytorch】关于y.backward()和x.grad2024-06-203.【深度学习】概率论知识复习2024-06-20

4.【DL】YOLOv8模型中的train.py2024-12-03

5.【DL】YOLO模型改进记录2024-12-036.【DL】深度学习环境配置（windows版本）2024-12-05

收起

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

最外层的train.py

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


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

models/yolo中的train.py

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

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

在这里调用这个变量

总结

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

补充1

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