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）
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|