18 lines
534 B
Python
18 lines
534 B
Python
#测试训练集
|
||
|
||
from ultralytics import YOLO
|
||
|
||
# 创建 YOLOv8 模型实例，选择适合的配置文件
|
||
model = YOLO('yolov8n.yaml') # YOLOv8 Nano 配置文件(这个是从官网上下载的预训练模型)
|
||
|
||
# 配置训练参数
|
||
model.train(
|
||
data='data.yaml', # 数据配置文件
|
||
epochs=150, # 训练轮数
|
||
batch=8, # 批量大小（适合 CPU 计算）
|
||
imgsz=320, # 图像尺寸（较小尺寸适合 CPU）
|
||
device='cpu' # 训练设备设置为 CPU（本人用的是cpu）
|
||
)
|
||
|
||
|