用于训练 YOLOv8 网络的射线调整检查点

Question

我一直在尝试用YOLOv8架构训练自己的微调网络，我也想优化超参数并找到数据增强的最佳参数。现在，我正在寻找可能的最佳参数，但由于训练持续了很多小时，所以我找不到如何从 YOLOv8 射线调整中保存检查点。原始 YOLOv8 Ray 调整文档中没有关于检查点的内容：YOLOv8 Ray Tune 文档。 可能有一种方法可以做到这一点，因为原始 Ray Tune 网页上有一个关于检查点的页面：原始文档，但我不知道如何在示例代码中实现它：

from ultralytics import YOLO

# Load a YOLOv8n model
model = YOLO('yolov8n.pt')

# Start tuning hyperparameters for YOLOv8n training on the COCO8 dataset
result_grid = model.tune(data='coco8.yaml', use_ray=True)

Answer 1

from ultralytics import YOLO

# Load a YOLOv8n model
model = YOLO('yolov8n.pt')

clsObject = ['persona', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic light','fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep','cow', 'elephant', 'bear', 'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee','skis', 'snowboard', 'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard', 'tennis racket', 'bottle', 'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple', 'sandwich', 'orange', 'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch', 'potted plant', 'bed',  'dining table', 'toilet', 'tv', 'laptop', 'mouse', 'remote', 'keyboard', 'cell phone', 'microwave', 'oven',    'toaster', 'sink', 'refrigerator', 'book', 'clock', 'vase', 'scissors', 'teddy bear', 'hair drier', 'toothbrush']

text_obj = f'{self.clsObject[cls]}'

size_obj, thickness_obj = 0.75, 1

frame = self.draw_text(frame, (0, 255, 0), text_obj, x1, y1, size_obj, thickness_obj, back=True)

frame = self.draw_area(frame, (0, 0, 255), x1, y1, x2, y2)