22 lines
717 B
Python
22 lines
717 B
Python
|
# coding:utf-8
|
|||
|
|
from ultralytics import YOLO
|
|||
|
|
|
|||
|
|
# 模型配置文件,此处以 s 为例，只需要写 yolo11s.yaml ，yolo11.yaml 会自动定位为 yolo11s 模型
|
|||
|
|
# 如果写成yolo11m.yaml ，yolo11.yaml 会自动定位为 yolo11m 模型
|
|||
|
|
model_yaml_path = r"ultralytics/cfg/models/11/MobileNetV4.yaml"
|
|||
|
|
# 数据集配置文件
|
|||
|
|
data_yaml_path = r'/home/hx/yolo11-jz/ailaidata.yaml'
|
|||
|
|
|
|||
|
|
# 预训练模型
|
|||
|
|
pre_model_name = 'yolo11n.pt'
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
# 加载预训练模型
|
|||
|
|
model = YOLO(model_yaml_path, task='detect').load(pre_model_name)
|
|||
|
|
# 训练模型
|
|||
|
|
results = model.train(
|
|||
|
|
data=data_yaml_path,
|
|||
|
|
epochs=100,
|
|||
|
|
batch=16,
|
|||
|
|
name='train_emoKDEF_MobileNetV4_s',
|
|||
|
|
)
|