文章目录
-
- 训练 --train.py
- 推理 --detect.py
- 验证 --val.py
- 不训练,只查看模型结构/参数量/计算量 --test.py
YOLOv10
是基于 YOLOv8
项目的改进版本,目前已经被 YOLOv8
项目合并,所以两个算法使用方法完全一致~
今天我给大家展示一种非常方便的使用过程,包含【训练】【验证】【推理】
不需要太复杂的代码更改,直接三个 python
文件对应三个功能 —【训练】【验证】【推理】
订阅过YOLOv8/v10
专栏的同学一定要先看这篇文章,可以让你少走很多弯路
训练 --train.py
大家可以在 ultralytics/ultralytics
文件夹下新建一个train.py
,然后直接复制我的代码,
所有训练有关的超参数都可以在这个文件中调节,基本和 YOLOv5
使用方式相同,
这里我说一下比较重要的几个参数:
YOLO( )
:这个参数里面可以写你模型yaml
文件的路径,也可以直接写.pt
文件YOLO( ).load( )
:如果你不加