Yolo11模型训练速通

Pycharm+python3.11+cuda设备：（应该用cpu也行，但是我没试过）

unity项目需要跑一个图像识别模型出来，对AI了解很少，python也用得不多，仅记录怎么快速跑通，不涉及调参数

一、yolo11

一个目标识别模型

 二、在pycharm进行环境配置

 需要torch、Ultralytics两个包

pycharm新建一个项目，良好的习惯是隔离开环境，因此建一个新环境是最好的（粗浅的理解就是新环境啥包也没安，避免版本冲突）

打开终端

 输入以下命令安装torch和ultralytics，严谨来说应该还要对一下cuda版本啥的，但是我直接这么安能正常运行，我就没管了

pip3 install torch torchvision torchaudio

pip install ultralytics

ps.安不上的找一下镜像源，我这里科学上网了（科学上网也得反复试几次才行）

三、写代码

from ultralytics import YOLOif __name__ == '__main__':# Load a modelmodel = YOLO("yolo11n.pt")  # load a pretrained model (recommended for training)# Train the model with GPUresults = model.train(data="coco8.yaml", epochs=100, imgsz=640, device=0)# Evaluate the model's performance on the validation setresults = model.val()# Export the model to ONNX formatsuccess = model.export(format="onnx")

coco8.yaml是训练集，可以换，另外最后一行导出ONNX格式是我个人需要， 可以不要。

运行后等它跑就行了。

运行完以后会输出这个，顺着地方找就能找到训练好的模型了。