Log In Sign Up
Models:
David Feaster
DavidFeaster/
Yolo-v8-Bleeding-Detectio
0
Downloads: 1
[ab8373]: / train_yolo_v8.py

Download this file

15 lines (11 with data), 209 Bytes 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
from ultralytics import YOLO







model = YOLO('yolov8x.pt')

model.to('cuda')



results = model.train(

    data="yolo.yaml",

    imgsz=224,

    epochs=500,

    batch=8,

    name='train_valid_4',

    patience=0,

)