YOLOv8项目中有两个地方可以执行train.py
最外层的train.py
(现在想起来这里好像是我自己写的训练脚本,原项目里没有)
YOLO()构造函数默认构造的是目标检测的模型,也可以在方法的参数中指定task类型
models/yolo中的train.py
这里的每个文件夹中定义了相对应任务中的训练脚本
在这里是通过这种方式获取任务类型的
在这里调用这个变量
总结
在yolo中,不同任务的选择是在创建模型时选择对应的任务模板来确定的
标签:DL,yolo,模型,py,YOLOv8,train From: https://www.cnblogs.com/satsuki26681534/p/18584931