Skip to main content

YOLO魔改

1.backbone添加

注意需要在源码上更改，使用源码添加软连接

ln -s /home/lxj/anaconda3/lib/python3.9/site-packages/ultralytics ultralytics

添加timm网络

https://github.com/huggingface/pytorch-image-models

2.输入改变

from ultralytics import YOLO

# Load a model
model = YOLO('yolov8n.yaml')  # build a new model from YAML
model = YOLO('yolov8n.pt')  # load a pretrained model (recommended for training)
model = YOLO('yolov8n.yaml').load('yolov8n.pt')  # build from YAML and transfer weights

# Train the model
model.train(data='coco128.yaml', epochs=100, imgsz=640, ch=1)

3.

1.backbone添加
2.输入改变
3.