13 lines
443 B
Python
13 lines
443 B
Python
from ultralytics import YOLO
|
|
|
|
def main():
|
|
# 加载模型
|
|
model = YOLO("/mnt/d/fiber/yolov8/ultralytics-main/ultralytics-main/ultralytics/cfg/models/v8/yolov8.yaml")
|
|
# 从头开始构建新模型
|
|
model = YOLO("yolov8x.pt") # 加载预训练模型
|
|
data_yaml = "coco8.yaml"
|
|
# 使用模型 imgsz 2144
|
|
model.train(data=data_yaml, epochs=500, batch=6, imgsz=2144,device=[0, 1, 2], patience=0)
|
|
|
|
if __name__ == '__main__':
|
|
main() |